From 11cb839b4aa88a0baf3463451309c8f2b5275d29 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Fri, 9 Feb 2024 20:03:36 +0200 Subject: [PATCH] Update util.py --- animatediff/utils/util.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/animatediff/utils/util.py b/animatediff/utils/util.py index ab4b6c5..bd1483d 100644 --- a/animatediff/utils/util.py +++ b/animatediff/utils/util.py @@ -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