Skip to content

【bug】loading error about waifu_diffusion_v1_4 model #7

Open
@xyxxmb

Description

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]).

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions