lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-12 08:29:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

avoid some mysteries problems when using lots of python local delegations

Browse Source
This commit is contained in:
layerdiffusion
2024-08-19 09:47:04 -07:00
parent 0b70b7287c
commit 2f1d04759f
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
backend/patcher/lora.py
View File

@@ -40,6 +40,7 @@ def model_lora_keys_unet(model, key_map={}):
return get_function('model_lora_keys_unet')(model, key_map)
@torch.inference_mode()
def weight_decompose(dora_scale, weight, lora_diff, alpha, strength, computation_dtype):
# Modified from https://github.com/comfyanonymous/ComfyUI/blob/39f114c44bb99d4a221e8da451d4f2a20119c674/comfy/model_patcher.py#L33
@@ -63,6 +64,7 @@ def weight_decompose(dora_scale, weight, lora_diff, alpha, strength, computation
return weight
@torch.inference_mode()
def merge_lora_to_weight(patches, weight, key="online_lora", computation_dtype=torch.float32):
# Modified from https://github.com/comfyanonymous/ComfyUI/blob/39f114c44bb99d4a221e8da451d4f2a20119c674/comfy/model_patcher.py#L446
@@ -296,6 +298,7 @@ class LoraLoader:
self.dirty = True
return list(p)
@torch.inference_mode()
def refresh(self, target_device=None, offload_device=torch.device('cpu')):
if not self.dirty:
return