Skip to content

Commit 098a352

Browse files
Add warning for torch-directml usage (#10482)
Added a warning message about the state of torch-directml.
1 parent e86b79a commit 098a352

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

comfy/model_management.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,7 @@ def get_supported_float8_types():
8989

9090
directml_enabled = False
9191
if args.directml is not None:
92+
logging.warning("WARNING: torch-directml barely works, is very slow, has not been updated in over 1 year and might be removed soon, please don't use it, there are better options.")
9293
import torch_directml
9394
directml_enabled = True
9495
device_index = args.directml

0 commit comments

Comments
 (0)