Skip to content

RuntimeError: Error(s) in loading state_dict for CRSControlNet #12

@lpy29

Description

@lpy29

Hello, I encountered the following error when running the inference.py:
DiffusionWrapper has 859.52 M params.
making attention of type 'vanilla' with 512 in_channels
Working with z of shape (1, 4, 32, 32) = 4096 dimensions.
making attention of type 'vanilla' with 512 in_channels
Loaded model config from [./configs/crs.yaml]
Loaded state_dict from [ckpts/last.ckpt]
Traceback (most recent call last):
File "src/test/inference.py", line 38, in
model.load_state_dict(load_state_dict('ckpts/last.ckpt', location='cuda'))
File "/opt/conda/envs/control-v11/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1604, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for CRSControlNet:
Missing key(s) in state_dict: "cond_stage_model.transformer.text_model.embeddings.position_ids".

Metadata

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