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

Commit 9c832aa

Browse files
committed
don't append cond_stage_model to CLIP
1 parent 6a7077c commit 9c832aa

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

utils/convert_diffusers_to_original_ms_text_to_video.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -424,10 +424,10 @@ def convert_text_enc_state_dict(text_enc_dict):
424424
# Need to add the tag 'transformer' in advance so we can knock it out from the final layer-norm
425425
text_enc_dict = {"transformer." + k: v for k, v in text_enc_dict.items()}
426426
text_enc_dict = convert_text_enc_state_dict_v20(text_enc_dict)
427-
text_enc_dict = {"cond_stage_model.model." + k: v for k, v in text_enc_dict.items()}
427+
#text_enc_dict = {"cond_stage_model.model." + k: v for k, v in text_enc_dict.items()}
428428
else:
429429
text_enc_dict = convert_text_enc_state_dict(text_enc_dict)
430-
text_enc_dict = {"cond_stage_model.transformer." + k: v for k, v in text_enc_dict.items()}
430+
#text_enc_dict = {"cond_stage_model.transformer." + k: v for k, v in text_enc_dict.items()}
431431

432432
# DON'T PUT TOGETHER FOR THE NEW CHECKPOINT AS MODELSCOPE USES THEM IN THE SPLITTED FORM --kabachuha
433433
# Save CLIP and the Diffusion model to their own files

0 commit comments

Comments
 (0)