wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-07 02:07:32 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

Fix cuda getting initialized in cpu mode. (#12779)

Browse Source
This commit is contained in:
comfyanonymous 2026-03-04 23:39:51 -08:00 committed by GitHub
parent c5fe8ace68
commit 4941671b5a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/model_management.py
View File

@ -1666,12 +1666,16 @@ def lora_compute_dtype(device):
return dtype return dtype
def synchronize(): def synchronize():
if cpu_mode():
return
if is_intel_xpu(): if is_intel_xpu():
torch.xpu.synchronize() torch.xpu.synchronize()
elif torch.cuda.is_available(): elif torch.cuda.is_available():
torch.cuda.synchronize() torch.cuda.synchronize()
def soft_empty_cache(force=False): def soft_empty_cache(force=False):
if cpu_mode():
return
global cpu_state global cpu_state
if cpu_state == CPUState.MPS: if cpu_state == CPUState.MPS:
torch.mps.empty_cache() torch.mps.empty_cache()