Add warning for torch-directml usage (#10482)

Added a warning message about the state of torch-directml.
This commit is contained in:
comfyanonymous
2025-10-25 17:05:22 -07:00
committed by GitHub
parent e86b79ab9e
commit 098a352f13

View File

@@ -89,6 +89,7 @@ if args.deterministic:
directml_enabled = False directml_enabled = False
if args.directml is not None: if args.directml is not None:
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.")
import torch_directml import torch_directml
directml_enabled = True directml_enabled = True
device_index = args.directml device_index = args.directml