Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

try fix lora vram (#357)

Browse Source 
* try fix lora vram

* try fix lora vram
This commit is contained in:
lllyasviel
2023-09-13 02:29:43 -07:00
committed by GitHub
parent eccf32b78c
commit a9b7219604
4 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/default_pipeline.py
+4
View File
@@ -106,6 +106,7 @@ refresh_loras([(modules.path.default_lora_name, 0.5), ('None', 0.5), ('None', 0.
expansion = FooocusExpansion()
@torch.no_grad()
def clip_encode_single(clip, text, verbose=False):
cached = clip.fcs_cond_cache.get(text, None)
if cached is not None:
@@ -120,6 +121,7 @@ def clip_encode_single(clip, text, verbose=False):
return result
@torch.no_grad()
def clip_encode(sd, texts, pool_top_k=1):
if sd is None:
return None
@@ -143,6 +145,7 @@ def clip_encode(sd, texts, pool_top_k=1):
return [[torch.cat(cond_list, dim=1), {"pooled_output": pooled_acc}]]
@torch.no_grad()
def clear_sd_cond_cache(sd):
if sd is None:
return None
@@ -152,6 +155,7 @@ def clear_sd_cond_cache(sd):
return
@torch.no_grad()
def clear_all_caches():
clear_sd_cond_cache(xl_base_patched)
clear_sd_cond_cache(xl_refiner)