xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-13 18:47:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

cuda

Browse Source
This commit is contained in:
camenduru 2024-08-08 12:52:42 +03:00 committed by GitHub
parent ace7627a86
commit 461d4ff420
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
totoro/model_management.py
View File

@ -515,7 +515,7 @@ def unet_offload_device():
if vram_state == VRAMState.HIGH_VRAM: if vram_state == VRAMState.HIGH_VRAM:
return get_torch_device() return get_torch_device()
else: else:
return torch.device("cpu") return torch.device("cuda")
def unet_inital_load_device(parameters, dtype): def unet_inital_load_device(parameters, dtype):
torch_dev = get_torch_device() torch_dev = get_torch_device()