Skip to content
This repository was archived by the owner on Dec 14, 2023. It is now read-only.

Commit b125d32

Browse files
committed
don't append 'model.diffusion_model.'
another artifact from orig stable diffusion
1 parent 50dff38 commit b125d32

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

utils/convert_diffusers_to_original_ms_text_to_video.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -391,7 +391,7 @@ def convert_text_enc_state_dict(text_enc_dict):
391391

392392
# Convert the UNet model
393393
unet_state_dict = convert_unet_state_dict(unet_state_dict)
394-
unet_state_dict = {"model.diffusion_model." + k: v for k, v in unet_state_dict.items()}
394+
#unet_state_dict = {"model.diffusion_model." + k: v for k, v in unet_state_dict.items()}
395395

396396
# Convert the VAE model
397397
# vae_state_dict = convert_vae_state_dict(vae_state_dict)

0 commit comments

Comments
 (0)