wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-16 00:12:33 +08:00
Code Issues Actions 22 Packages Projects Releases Wiki Activity

To enable support for non-CUDA devices, substitute torch.cuda.empty_cache() with model_management.soft_empty_cache().

Browse Source
This commit is contained in:
KBRASK 2025-04-20 15:47:18 +08:00
parent 4d4e6a05fa
commit 0264149f46
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/ldm/modules/diffusionmodules/model.py
View File

@ -721,13 +721,13 @@ class Decoder(nn.Module):
if len(self.up[i_level].attn) > 0: if len(self.up[i_level].attn) > 0:
h_new = self.up[i_level].attn[i_block](h_new, **kwargs) h_new = self.up[i_level].attn[i_block](h_new, **kwargs)
del h del h
torch.cuda.empty_cache() model_management.soft_empty_cache()
h = h_new h = h_new
if i_level != 0: if i_level != 0:
h_new = self.up[i_level].upsample(h) h_new = self.up[i_level].upsample(h)
del h del h
torch.cuda.empty_cache() model_management.soft_empty_cache()
h = h_new h = h_new
# end # end