lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-30 11:11:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

always set empty cache signal as long as any patch happens

Browse Source
This commit is contained in:
layerdiffusion
2024-08-23 08:56:57 -07:00
parent df598c4df5
commit f23ee63cb3
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/patcher/lora.py
View File

@@ -346,6 +346,8 @@ class LoraLoader:
# Patch # Patch
memory_management.signal_empty_cache = True
for key, current_patches in (tqdm(self.patches.items(), desc=f'Patching LoRAs for {type(self.model).__name__}') if len(self.patches) > 0 else self.patches): for key, current_patches in (tqdm(self.patches.items(), desc=f'Patching LoRAs for {type(self.model).__name__}') if len(self.patches) > 0 else self.patches):
try: try:
parent_layer, child_key, weight = utils.get_attr_with_parent(self.model, key) parent_layer, child_key, weight = utils.get_attr_with_parent(self.model, key)