mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-11 06:40:48 +08:00
Fix directml again now that I understand what the command line is doing
This commit is contained in:
parent
09838ed604
commit
3367362cec
@ -119,6 +119,7 @@ parser.add_argument("--plausible-analytics-domain", required=False,
|
||||
help="Specifies the domain name for analytics events.")
|
||||
parser.add_argument("--analytics-use-identity-provider", action="store_true",
|
||||
help="Uses platform identifiers for unique visitor analytics.")
|
||||
|
||||
if options.args_parsing:
|
||||
args, _ = parser.parse_known_args()
|
||||
else:
|
||||
|
||||
@ -37,7 +37,7 @@ if args.deterministic:
|
||||
torch.use_deterministic_algorithms(True, warn_only=True)
|
||||
|
||||
directml_enabled = False
|
||||
if args.directml:
|
||||
if args.directml is not None:
|
||||
import torch_directml
|
||||
directml_enabled = True
|
||||
device_index = args.directml
|
||||
|
||||
Loading…
Reference in New Issue
Block a user