T2I-Adapter icon indicating copy to clipboard operation
T2I-Adapter copied to clipboard

How to load the trained model?

Open 1llss opened this issue 11 months ago • 1 comments

I use the train_sketch.py ​​file given by the author to train on my own data set. The trained model weight file is a .pth file. The following error occurred when I loaded my trained model weights: Traceback (most recent call last): File "test.py", line 19, in adapter.load_state_dict(state_dict) File "/opt/data/private/T2I-Adapter-XL/venv/lib/python3.8/site-packages/torch/nn/modules/module.py", line 2041, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for T2IAdapter: Missing key(s) in state_dict: "adapter.conv_in.weight", "adapter.conv_in.bias", "adapter.body.0.resnets.0.block1.weight", "adapter.body.0.resnets.0.block1.bias", "adapter.body.0.resnets.0.block2.weight", "adapter.body.0.resnets.0.block2.bias", "adapter.body.0.resnets.1.block1.weight", "adapter.body.0.resnets.1.block1.bias", "adapter.body.0.resnets.1.block2.weight", "adapter.body.0.resnets.1.block2.bias", "adapter.body.1.in_conv.weight", "adapter.body.1.in_conv.bias", "adapter.body.1.resnets.0.block1.weight", "adapter.body.1.resnets.0.block1.bias", "adapter.body.1.resnets.0.block2.weight", "adapter.body.1.resnets.0.block2.bias", "adapter.body.1.resnets.1.block1.weight", "adapter.body.1.resnets.1.block1.bias", "adapter.body.1.resnets.1.block2.weight", "adapter.body.1.resnets.1.block2.bias", "adapter.body.2.in_conv.weight", "adapter.body.2.in_conv.bias", "adapter.body.2.resnets.0.block1.weight", "adapter.body.2.resnets.0.block1.bias", "adapter.body.2.resnets.0.block2.weight", "adapter.body.2.resnets.0.block2.bias", "adapter.body.2.resnets.1.block1.weight", "adapter.body.2.resnets.1.block1.bias", "adapter.body.2.resnets.1.block2.weight", "adapter.body.2.resnets.1.block2.bias", "adapter.body.3.resnets.0.block1.weight", "adapter.body.3.resnets.0.block1.bias", "adapter.body.3.resnets.0.block2.weight", "adapter.body.3.resnets.0.block2.bias", "adapter.body.3.resnets.1.block1.weight", "adapter.body.3.resnets.1.block1.bias", "adapter.body.3.resnets.1.block2.weight", "adapter.body.3.resnets.1.block2.bias". Unexpected key(s) in state_dict: "body.0.block1.weight", "body.0.block1.bias", "body.0.block2.weight", "body.0.block2.bias", "body.1.block1.weight", "body.1.block1.bias", "body.1.block2.weight", "body.1.block2.bias", "body.2.in_conv.weight", "body.2.in_conv.bias", "body.2.block1.weight", "body.2.block1.bias", "body.2.block2.weight", "body.2.block2.bias", "body.3.block1.weight", "body.3.block1.bias", "body.3.block2.weight", "body.3.block2.bias", "body.4.in_conv.weight", "body.4.in_conv.bias", "body.4.block1.weight", "body.4.block1.bias", "body.4.block2.weight", "body.4.block2.bias", "body.5.block1.weight", "body.5.block1.bias", "body.5.block2.weight", "body.5.block2.bias", "body.6.block1.weight", "body.6.block1.bias", "body.6.block2.weight", "body.6.block2.bias", "body.7.block1.weight", "body.7.block1.bias", "body.7.block2.weight", "body.7.block2.bias", "conv_in.weight", "conv_in.bias"

image image

1llss avatar Feb 27 '24 17:02 1llss

Have you ever solved the problem? Does it mean diffuser pipe not support testing from self-training pth files? Maybe we have to change the key mentioned above.

JinShuwenABC avatar Mar 21 '24 14:03 JinShuwenABC