wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-27 17:37:39 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

Merge 1c9f5960f3 into b30e980a20

Browse Source
This commit is contained in:
Rui Wang (王瑞) 2026-05-24 16:00:28 -07:00 committed by GitHub
commit 04fcd8c31d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/model_management.py
View File

@ -131,6 +131,7 @@ except:
try: try:
import torch_npu # noqa: F401 import torch_npu # noqa: F401
_ = torch.npu.device_count() _ = torch.npu.device_count()
torch_npu.npu.set_compile_mode(jit_compile = False)
npu_available = torch.npu.is_available() npu_available = torch.npu.is_available()
except: except:
npu_available = False npu_available = False