|
|
|
@ -55,13 +55,26 @@ def load_clip_weights(model, sd):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def load_lora_for_models(model, clip, lora, strength_model, strength_clip): |
|
|
|
|
key_map = comfy.lora.model_lora_keys_unet(model.model) |
|
|
|
|
key_map = comfy.lora.model_lora_keys_clip(clip.cond_stage_model, key_map) |
|
|
|
|
key_map = {} |
|
|
|
|
if model is not None: |
|
|
|
|
key_map = comfy.lora.model_lora_keys_unet(model.model, key_map) |
|
|
|
|
if clip is not None: |
|
|
|
|
key_map = comfy.lora.model_lora_keys_clip(clip.cond_stage_model, key_map) |
|
|
|
|
|
|
|
|
|
loaded = comfy.lora.load_lora(lora, key_map) |
|
|
|
|
new_modelpatcher = model.clone() |
|
|
|
|
k = new_modelpatcher.add_patches(loaded, strength_model) |
|
|
|
|
new_clip = clip.clone() |
|
|
|
|
k1 = new_clip.add_patches(loaded, strength_clip) |
|
|
|
|
if model is not None: |
|
|
|
|
new_modelpatcher = model.clone() |
|
|
|
|
k = new_modelpatcher.add_patches(loaded, strength_model) |
|
|
|
|
else: |
|
|
|
|
k = () |
|
|
|
|
new_modelpatcher = None |
|
|
|
|
|
|
|
|
|
if clip is not None: |
|
|
|
|
new_clip = clip.clone() |
|
|
|
|
k1 = new_clip.add_patches(loaded, strength_clip) |
|
|
|
|
else: |
|
|
|
|
k1 = () |
|
|
|
|
new_clip = None |
|
|
|
|
k = set(k) |
|
|
|
|
k1 = set(k1) |
|
|
|
|
for x in loaded: |
|
|
|
|