fix a small bug

2023-12-25 14:27:57 +08:00 · 2023-12-25 14:27:57 +08:00 · 1c3c9908e0
parent 166c54c493
commit 1c3c9908e0
1 changed files with 1 additions and 1 deletions
--- a/tutorial_train_faceid.py
+++ b/tutorial_train_faceid.py
@ -319,7 +319,7 @@ def main():
                "to_v_ip.weight": unet_sd[layer_name + ".to_v.weight"],
            }
            attn_procs[name] = LoRAIPAttnProcessor(hidden_size=hidden_size, cross_attention_dim=cross_attention_dim, rank=lora_rank)
-            attn_procs[name].load_state_dict(weights)
+            attn_procs[name].load_state_dict(weights, strict=False)
    unet.set_attn_processor(attn_procs)
    adapter_modules = torch.nn.ModuleList(unet.attn_processors.values())