Skip to content

GitLab

"docs/vscode:/vscode.git/clone" did not exist on "8369196703e07a42e7835a65b223c42d4e993276"
Commit 727588d0 authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix some new loras.

parent 315ba30c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
comfy/sd.py
View file @ 727588d0
...@@ -170,6 +170,8 @@ def model_lora_keys_clip(model, key_map={}): ...@@ -170,6 +170,8 @@ def model_lora_keys_clip(model, key_map={}):
if k in sdk: if k in sdk:
lora_key = text_model_lora_key.format(b, LORA_CLIP_MAP[c]) lora_key = text_model_lora_key.format(b, LORA_CLIP_MAP[c])
key_map[lora_key] = k key_map[lora_key] = k
lora_key = "lora_te1_text_model_encoder_layers_{}_{}".format(b, LORA_CLIP_MAP[c])
key_map[lora_key] = k
k = "clip_l.transformer.text_model.encoder.layers.{}.{}.weight".format(b, c) k = "clip_l.transformer.text_model.encoder.layers.{}.{}.weight".format(b, c)
if k in sdk: if k in sdk:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment