wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-31 08:40:19 +08:00
Code Issues Actions 26 Packages Projects Releases Wiki Activity

Fix args.directml is not None error

Browse Source
This commit is contained in:
Benjamin Berman 2024-02-08 08:40:13 -08:00
parent b8fc850b47
commit 8508a5a853
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_management.py
View File

@ -37,7 +37,7 @@ if args.deterministic:
torch.use_deterministic_algorithms(True, warn_only=True) torch.use_deterministic_algorithms(True, warn_only=True)
directml_enabled = False directml_enabled = False
if args.directml is not None: if args.directml:
import torch_directml import torch_directml
directml_enabled = True directml_enabled = True
device_index = args.directml device_index = args.directml