Update util.py

2024-02-09 20:03:36 +02:00
parent 555af1dba0
commit 11cb839b4a
1 changed files with 1 additions and 1 deletions
@@ -135,7 +135,7 @@ def load_weights(
        converted_unet_checkpoint = convert_ldm_unet_checkpoint(dreambooth_state_dict, animation_pipeline.unet.config)
        animation_pipeline.unet.load_state_dict(converted_unet_checkpoint, strict=False)
        # 3. text_model
-        animation_pipeline.text_encoder = convert_ldm_clip_checkpoint(dreambooth_state_dict)
+        animation_pipeline.text_encoder = convert_ldm_clip_checkpoint(dreambooth_state_dict, strict=False)
        del dreambooth_state_dict
        
    # lora layers