-
Notifications
You must be signed in to change notification settings - Fork 8
RuntimeError: Error(s) in loading state_dict for CRSControlNet #12
Description
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".