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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -89,6 +89,7 @@ if args.deterministic:
directml_enabled = False
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
directml_enabled = True
device_index = args.directml