search

camenduru / controlnet-colab

The Unlicense
519 stars 59 forks source link

【bug】loading error about waifu_diffusion_v1_4 model #7

Open xyxxmb opened 1 year ago

xyxxmb commented 1 year ago 
Loaded model config from [ControlNet/models/cldm_v15.yaml]
Loaded state_dict from [models/wd-1-4-anime_e1.ckpt]
Traceback (most recent call last):
  File "app.py", line 52, in <module>
    model = Model()
  File "/content/ControlNet-with-other-models/model.py", line 67, in __init__
    self.set_base_model(DEFAULT_BASE_MODEL_REPO,
  File "/content/ControlNet-with-other-models/model.py", line 75, in set_base_model
    self.load_base_model(base_model_url)
  File "/content/ControlNet-with-other-models/model.py", line 89, in load_base_model
    self.model.load_state_dict(load_state_dict(model_path,
  File "/content/anaconda3/envs/control/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1671, in load_state_dict
    raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for ControlLDM:
    size mismatch for model.diffusion_model.input_blocks.1.1.proj_in.weight: copying a param with shape torch.Size([320, 320]) from checkpoint, the shape in current model is torch.Size([320, 320, 1, 1]).
    size mismatch for model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn2.to_k.weight: copying a param with shape torch.Size([320, 1024]) from checkpoint, the shape in current model is torch.Size([320, 768]).
    size mismatch for model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn2.to_v.weight: copying a param with shape torch.Size([320, 1024]) from checkpoint, the shape in current model is torch.Size([320, 768]).