KeyError: 'cond_stage_model.transformer.text_model.embeddings.token_embedding.weight'

[Carlyx]

custom-diffusion/train.py

Line 765 in 51a498f

token_weights = st["cond_stage_model.transformer.text_model.embeddings.token_embedding.weight"]

And I run :

bash scripts/finetune_real.sh "cat" data/cat real_reg/samples_cat  cat finetune_addtoken.yaml delta_cat.ckpt

[Carlyx]

And these are keys of st:

[nupurkmr9]

Hi, as we fine-tune the pretrained stable diffusion checkpoint. The following below commands should work:

wget https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt 
bash scripts/finetune_real.sh "cat" data/cat real_reg/samples_cat  cat finetune_addtoken.yaml sd-v1-4.ckpt

Thanks.