fix for flux clip

2025-02-01 17:45:00 +09:00 · 2025-02-01 17:45:00 +09:00 · ac0f869332
parent 14961b874a
commit ac0f869332
1 changed files with 3 additions and 1 deletions
--- a/scripts/lora_block_weight.py
+++ b/scripts/lora_block_weight.py
@ -1311,8 +1311,10 @@ def get_flux_blocks(key):
        return "VAE"
    if "t5xxl" in key:
        return "T5"
-    if "text_encoders.clip" in key:
+    if "clip_" in key:
        return "CLIP"
+    if "t5xxl" in key:
+        return "T5"
    
    match = re.search(r'\_(\d+)\_', key)
    if "double_blocks" in key: