Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

【bug】loading error about waifu_diffusion_v1_4 model #7

Open
xyxxmb opened this issue Mar 3, 2023 · 0 comments
Open

【bug】loading error about waifu_diffusion_v1_4 model #7

xyxxmb opened this issue Mar 3, 2023 · 0 comments

Comments

@xyxxmb
Copy link

xyxxmb commented Mar 3, 2023

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]).
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant