mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-15 05:57:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Some fixes to generalize CUDA specific functionality to Intel or other GPUs.

Browse Source
This commit is contained in:
Simon Lui
2023-09-02 18:22:10 -07:00
parent 62efc78a4b
commit 4a0c4ce4ef
3 changed files with 38 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/ldm/modules/attention.py
View File

@@ -323,8 +323,7 @@ class CrossAttentionDoggettx(nn.Module):
break
except model_management.OOM_EXCEPTION as e:
if first_op_done == False:
torch.cuda.empty_cache()
torch.cuda.ipc_collect()
model_management.soft_empty_cache()
if cleared_cache == False:
cleared_cache = True
print("out of memory error, emptying cache and trying again")