search

cripac-sjx / Anyface

Official implemention of Anyface++
7 stars 0 forks source link

Missing key(s) in StyleCLIPMapper When loading model #2

Open Jacky-hate opened 3 months ago

Jacky-hate commented 3 months ago

When I run the script:

python scripts/synthesis.py --descrip "A girl with curly black hair is smiling." \
--memory_path "<path to memory model>" \
--checkpoints_path "<path to checkpoint path>" \
--exp_dir "outputs"

It shows an error:

Traceback (most recent call last):
  File "synthesis.py", line 76, in <module>
    run(test_opts)
  File "synthesis.py", line 43, in run
    net.load_state_dict(ckpt['state_dict'])
  File "/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1672, in load_state_dict
    self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for StyleCLIPMapper:
        Missing key(s) in state_dict: "CA_NET.fc.0.weight", "CA_NET.fc.0.bias", "CA_NET.fc.1.weight", "CA_NET.fc.1.bias", "CA_NET.fc.2.weight", "CA_NET.fc.2.bias", "CA_NET.fc.3.weight", "CA_NET.fc.3.bias", "T_map.transformer_encoder.layers.0.self_attn.in_proj_weight", "T_map.transformer_encoder.layers.0.self_attn.in_proj_bias", "T_map.transformer_encoder.layers.0.self_attn.out_proj.weight", "T_map.transformer_encoder.layers.0.self_attn.out_proj.bias", "T_map.transformer_encoder.layers.0.linear1.weight", "T_map.transformer_encoder.layers.0.linear1.bias", "T_map.transformer_encoder.layers.0.linear2.weight", "T_map.transformer_encoder.layers.0.linear2.bias", "T_map.transformer_encoder.layers.0.norm1.weight", "T_map.transformer_encoder.layers.0.norm1.bias", "T_map.transformer_encoder.layers.0.norm2.weight", "T_map.transformer_encoder.layers.0.norm2.bias", "T_map.transformer_encoder.layers.1.self_attn.in_proj_weight", "T_map.transformer_encoder.layers.1.self_attn.in_proj_bias", "T_map.transformer_encoder.layers.1.self_attn.out_proj.weight", "T_map.transformer_encoder.layers.1.self_attn.out_proj.bias", "T_map.transformer_encoder.layers.1.linear1.weight", "T_map.transformer_encoder.layers.1.linear1.bias", "T_map.transformer_encoder.layers.1.linear2.weight", "T_map.transformer_encoder.layers.1.linear2.bias", "T_map.transformer_encoder.layers.1.norm1.weight", "T_map.transformer_encoder.layers.1.norm1.bias", "T_map.transformer_encoder.layers.1.norm2.weight", "T_map.transformer_encoder.layers.1.norm2.bias", "T_map.transformer_encoder.layers.2.self_attn.in_proj_weight", "T_map.transformer_encoder.layers.2.self_attn.in_proj_bias", "T_map.transformer_encoder.layers.2.self_attn.out_proj.weight", "T_map.transformer_encoder.layers.2.self_attn.out_proj.bias", "T_map.transformer_encoder.layers.2.linear1.weight", "T_map.transformer_encoder.layers.2.linear1.bias", "T_map.transformer_encoder.layers.2.linear2.weight", "T_map.transformer_encoder.layers.2.linear2.bias", "T_map.transformer_encoder.layers.2.norm1.weight", "T_map.transformer_encoder.layers.2.norm1.bias", "T_map.transformer_encoder.layers.2.norm2.weight", "T_map.transformer_encoder.layers.2.norm2.bias", "T_map.transformer_encoder.layers.3.self_attn.in_proj_weight", "T_map.transformer_encoder.layers.3.self_attn.in_proj_bias", "T_map.transformer_encoder.layers.3.self_attn.out_proj.weight", "T_map.transformer_encoder.layers.3.self_attn.out_proj.bias", "T_map.transformer_encoder.layers.3.linear1.weight", "T_map.transformer_encoder.layers.3.linear1.bias", "T_map.transformer_encoder.layers.3.linear2.weight", "T_map.transformer_encoder.layers.3.linear2.bias", "T_map.transformer_encoder.layers.3.norm1.weight", "T_map.transformer_encoder.layers.3.norm1.bias", "T_map.transformer_encoder.layers.3.norm2.weight", "T_map.transformer_encoder.layers.3.norm2.bias", "T_map.transformer_encoder.layers.4.self_attn.in_proj_weight", "T_map.transformer_encoder.layers.4.self_attn.in_proj_bias", "T_map.transformer_encoder.layers.4.self_attn.out_proj.weight", "T_map.transformer_encoder.layers.4.self_attn.out_proj.bias", "T_map.transformer_encoder.layers.4.linear1.weight", "T_map.transformer_encoder.layers.4.linear1.bias", "T_map.transformer_encoder.layers.4.linear2.weight", "T_map.transformer_encoder.layers.4.linear2.bias", "T_map.transformer_encoder.layers.4.norm1.weight", "T_map.transformer_encoder.layers.4.norm1.bias", "T_map.transformer_encoder.layers.4.norm2.weight", "T_map.transformer_encoder.layers.4.norm2.bias", "T_map.transformer_encoder.layers.5.self_attn.in_proj_weight", "T_map.transformer_encoder.layers.5.self_attn.in_proj_bias", "T_map.transformer_encoder.layers.5.self_attn.out_proj.weight", "T_map.transformer_encoder.layers.5.self_attn.out_proj.bias", "T_map.transformer_encoder.layers.5.linear1.weight", "T_map.transformer_encoder.layers.5.linear1.bias", "T_map.transformer_encoder.layers.5.linear2.weight", "T_map.transformer_encoder.layers.5.linear2.bias", "T_map.transformer_encoder.layers.5.norm1.weight", "T_map.transformer_encoder.layers.5.norm1.bias", "T_map.transformer_encoder.layers.5.norm2.weight", "T_map.transformer_encoder.layers.5.norm2.bias", "T_map.transformer_encoder.layers.6.self_attn.in_proj_weight", "T_map.transformer_encoder.layers.6.self_attn.in_proj_bias", "T_map.transformer_encoder.layers.6.self_attn.out_proj.weight", "T_map.transformer_encoder.layers.6.self_attn.out_proj.bias", "T_map.transformer_encoder.layers.6.linear1.weight", "T_map.transformer_encoder.layers.6.linear1.bias", "T_map.transformer_encoder.layers.6.linear2.weight", "T_map.transformer_encoder.layers.6.linear2.bias", "T_map.transformer_encoder.layers.6.norm1.weight", "T_map.transformer_encoder.layers.6.norm1.bias", "T_map.transformer_encoder.layers.6.norm2.weight", "T_map.transformer_encoder.layers.6.norm2.bias", "T_map.transformer_encoder.layers.7.self_attn.in_proj_weight", "T_map.transformer_encoder.layers.7.self_attn.in_proj_bias", "T_map.transformer_encoder.layers.7.self_attn.out_proj.weight", "T_map.transformer_encoder.layers.7.self_attn.out_proj.bias", "T_map.transformer_encoder.layers.7.linear1.weight", "T_map.transformer_encoder.layers.7.linear1.bias", "T_map.transformer_encoder.layers.7.linear2.weight", "T_map.transformer_encoder.layers.7.linear2.bias", "T_map.transformer_encoder.layers.7.norm1.weight", "T_map.transformer_encoder.layers.7.norm1.bias", "T_map.transformer_encoder.layers.7.norm2.weight", "T_map.transformer_encoder.layers.7.norm2.bias", "T_map.transformer_encoder.layers.8.self_attn.in_proj_weight", "T_map.transformer_encoder.layers.8.self_attn.in_proj_bias", "T_map.transformer_encoder.layers.8.self_attn.out_proj.weight", "T_map.transformer_encoder.layers.8.self_attn.out_proj.bias", "T_map.transformer_encoder.layers.8.linear1.weight", "T_map.transformer_encoder.layers.8.linear1.bias", "T_map.transformer_encoder.layers.8.linear2.weight", "T_map.transformer_encoder.layers.8.linear2.bias", "T_map.transformer_encoder.layers.8.norm1.weight", "T_map.transformer_encoder.layers.8.norm1.bias", "T_map.transformer_encoder.layers.8.norm2.weight", "T_map.transformer_encoder.layers.8.norm2.bias", "T_map.transformer_encoder.layers.9.self_attn.in_proj_weight", "T_map.transformer_encoder.layers.9.self_attn.in_proj_bias", "T_map.transformer_encoder.layers.9.self_attn.out_proj.weight", "T_map.transformer_encoder.layers.9.self_attn.out_proj.bias", "T_map.transformer_encoder.layers.9.linear1.weight", "T_map.transformer_encoder.layers.9.linear1.bias", "T_map.transformer_encoder.layers.9.linear2.weight", "T_map.transformer_encoder.layers.9.linear2.bias", "T_map.transformer_encoder.layers.9.norm1.weight", "T_map.transformer_encoder.layers.9.norm1.bias", "T_map.transformer_encoder.layers.9.norm2.weight", "T_map.transformer_encoder.layers.9.norm2.bias", "T_map.transformer_encoder.layers.10.self_attn.in_proj_weight", "T_map.transformer_encoder.layers.10.self_attn.in_proj_bias", "T_map.transformer_encoder.layers.10.self_attn.out_proj.weight", "T_map.transformer_encoder.layers.10.self_attn.out_proj.bias", "T_map.transformer_encoder.layers.10.linear1.weight", "T_map.transformer_encoder.layers.10.linear1.bias", "T_map.transformer_encoder.layers.10.linear2.weight", "T_map.transformer_encoder.layers.10.linear2.bias", "T_map.transformer_encoder.layers.10.norm1.weight", "T_map.transformer_encoder.layers.10.norm1.bias", "T_map.transformer_encoder.layers.10.norm2.weight", "T_map.transformer_encoder.layers.10.norm2.bias", "T_map.transformer_encoder.layers.11.self_attn.in_proj_weight", "T_map.transformer_encoder.layers.11.self_attn.in_proj_bias", "T_map.transformer_encoder.layers.11.self_attn.out_proj.weight", "T_map.transformer_encoder.layers.11.self_attn.out_proj.bias", "T_map.transformer_encoder.layers.11.linear1.weight", "T_map.transformer_encoder.layers.11.linear1.bias", "T_map.transformer_encoder.layers.11.linear2.weight", "T_map.transformer_encoder.layers.11.linear2.bias", "T_map.transformer_encoder.layers.11.norm1.weight", "T_map.transformer_encoder.layers.11.norm1.bias", "T_map.transformer_encoder.layers.11.norm2.weight", "T_map.transformer_encoder.layers.11.norm2.bias", "T_map.mapping.0.weight", "T_map.mapping.0.bias", "T_map.mapping.1.weight", "T_map.mapping.1.bias", "T_map.mapping.2.weight", "T_map.mapping.2.bias", "T_map.mapping.3.weight", "T_map.mapping.3.bias", "T_map.mapping.4.weight", "T_map.mapping.4.bias".

The model I used was downloaded from the google drive as mentioned by Readme.

YixuanGao98 commented 2 weeks ago

Same issue. Have you solved this issue?