Open
Description
Describe the bug
hi. I have tried to convert the cosmos-transfer1's base model to diffuers using "convert_cosmos_to_diffusers.py" code with options --transformer_type Cosmo
s-1.0-Diffusion-7B-Video2World --vae_type CV8x8x8-1.0 --transformer_ckpt_path ../fsdp_edge_v1/iter_000016000_ema_model_only.pt --output_path ./convert_to_diffusers
but I got error
File "/home1/jovyan/workspace/cosmos-transfer1/diffusers/../convert_cosmos_to_diffusers.py", line 485, in <module>
transformer = convert_transformer(args.transformer_type, args.transformer_ckpt_path, weights_only)
File "/home1/jovyan/workspace/cosmos-transfer1/diffusers/../convert_cosmos_to_diffusers.py", line 358, in convert_transformer
transformer.load_state_dict(original_state_dict, strict=True, assign=True)
File "/opt/conda/envs/cosmos-transfer1/lib/python3.10/site-packages/torch/nn/modules/module.py", line 2581, in load_state_dict
raise RuntimeError(
RuntimeError: Error(s) in loading state_dict for CosmosTransformer3DModel:
Missing key(s) in state_dict: "transformer_blocks.3.norm1.linear_1.weight", "transformer_blocks.3.norm1.linear_2.weight", "transformer_blocks.3.attn1.norm_q.weight", "transformer_blocks.3.attn1.norm_k.weight", "transformer_blocks.3.attn1.to_q.weight", "transformer_blocks.3.attn1.to_k.weight", "transformer_blocks.3.attn1.to_v.weight", "transformer_blocks.3.attn1.to_out.0.weight", "transformer_blocks.3.norm2.linear_1.weight", "transformer_blocks.3.norm2.linear_2.weight", "transformer_blocks.3.attn2.norm_q.weight", "transformer_blocks.3.attn2.norm_k.weight", "transformer_blocks.3.attn2.to_q.weight", "transformer_blocks.3.attn2.to_k.weight", "transformer_blocks.3.attn2.to_v.weight", "transformer_blocks.3.attn2.to_out.0.weight", "transformer_blocks.3.norm3.linear_1.weight", "transformer_blocks.3.norm3.linear_2.weight", "transformer_blocks.3.ff.net.0.proj.weight", "transformer_blocks.3.ff.net.2.weight", "transformer_blocks.4.norm1.linear_1.weight", "transformer_blocks.4.norm1.linear_2.weight", "transformer_blocks.4.attn1.norm_q.weight", "transformer_blocks.4.attn1.norm_k.weight", "transformer_blocks.4.attn1.to_q.weight", "transformer_blocks.4.attn1.to_k.weight", "transformer_blocks.4.attn1.to_v.weight", "transformer_blocks.4.attn1.to_out.0.weight", "transformer_blocks.4.norm2.linear_1.weight", "transformer_blocks.4.norm2.linear_2.weight", "transformer_blocks.4.attn2.norm_q.weight", "transformer_blocks.4.attn2.norm_k.weight", "transformer_blocks.4.attn2.to_q.weight", "transformer_blocks.4.attn2.to_k.weight", "transformer_blocks.4.attn2.to_v.weight", "transformer_blocks.4.attn2.to_out.0.weight", "transformer_blocks.4.norm3.linear_1.weight", "transformer_blocks.4.norm3.linear_2.weight", "transformer_blocks.4.ff.net.0.proj.weight", "transformer_blocks.4.ff.net.2.weight", "transformer_blocks.5.norm1.linear_1.weight", "transformer_blocks.5.norm1.linear_2.weight", "transformer_blocks.5.attn1.norm_q.weight", "transformer_blocks.5.attn1.norm_k.weight", "transformer_blocks.5.attn1.to_q.weight", "transformer_blocks.5.attn1.to_k.weight", "transformer_blocks.5.attn1.to_v.weight", "transformer_blocks.5.attn1.to_out.0.weight", "transformer_blocks.5.norm2.linear_1.weight", "transformer_blocks.5.norm2.linear_2.weight", "transformer_blocks.5.attn2.norm_q.weight", "transformer_blocks.5.attn2.norm_k.weight", "transformer_blocks.5.attn2.to_q.weight", "transformer_blocks.5.attn2.to_k.weight", "transformer_blocks.5.attn2.to_v.weight", "transformer_blocks.5.attn2.to_out.0.weight", "transformer_blocks.5.norm3.linear_1.weight", "transformer_blocks.5.norm3.linear_2.weight", "transformer_blocks.5.ff.net.0.proj.weight", "transformer_blocks.5.ff.net.2.weight", "transformer_blocks.6.norm1.linear_1.weight", "transformer_blocks.6.norm1.linear_2.weight", "transformer_blocks.6.attn1.norm_q.weight", "transformer_blocks.6.attn1.norm_k.weight", "transformer_blocks.6.attn1.to_q.weight", "transformer_blocks.6.attn1.to_k.weight", "transformer_blocks.6.attn1.to_v.weight", "transformer_blocks.6.attn1.to_out.0.weight", "transformer_blocks.6.norm2.linear_1.weight", "transformer_blocks.6.norm2.linear_2.weight", "transformer_blocks.6.attn2.norm_q.weight", "transformer_blocks.6.attn2.norm_k.weight", "transformer_blocks.6.attn2.to_q.weight", "transformer_blocks.6.attn2.to_k.weight", "transformer_blocks.6.attn2.to_v.weight", "transformer_blocks.6.attn2.to_out.0.weight", "transformer_blocks.6.norm3.linear_1.weight", "transformer_blocks.6.norm3.linear_2.weight", "transformer_blocks.6.ff.net.0.proj.weight", "transformer_blocks.6.ff.net.2.weight", "transformer_blocks.7.norm1.linear_1.weight", "transformer_blocks.7.norm1.linear_2.weight", "transformer_blocks.7.attn1.norm_q.weight", "transformer_blocks.7.attn1.norm_k.weight", "transformer_blocks.7.attn1.to_q.weight", "transformer_blocks.7.attn1.to_k.weight", "transformer_blocks.7.attn1.to_v.weight", "transformer_blocks.7.attn1.to_out.0.weight", "transformer_blocks.7.norm2.linear_1.weight", "transformer_blocks.7.norm2.linear_2.weight", "transformer_blocks.7.attn2.norm_q.weight", "transformer_blocks.7.attn2.norm_k.weight", "transformer_blocks.7.attn2.to_q.weight", "transformer_blocks.7.attn2.to_k.weight", "transformer_blocks.7.attn2.to_v.weight", "transformer_blocks.7.attn2.to_out.0.weight", "transformer_blocks.7.norm3.linear_1.weight", "transformer_blocks.7.norm3.linear_2.weight", "transformer_blocks.7.ff.net.0.proj.weight", "transformer_blocks.7.ff.net.2.weight", "transformer_blocks.8.norm1.linear_1.weight", "transformer_blocks.8.norm1.linear_2.weight", "transformer_blocks.8.attn1.norm_q.weight", "transformer_blocks.8.attn1.norm_k.weight", "transformer_blocks.8.attn1.to_q.weight", "transformer_blocks.8.attn1.to_k.weight", "transformer_blocks.8.attn1.to_v.weight", "transformer_blocks.8.attn1.to_out.0.weight", "transformer_blocks.8.norm2.linear_1.weight", "transformer_blocks.8.norm2.linear_2.weight", "transformer_blocks.8.attn2.norm_q.weight", "transformer_blocks.8.attn2.norm_k.weight", "transformer_blocks.8.attn2.to_q.weight", "transformer_blocks.8.attn2.to_k.weight", "transformer_blocks.8.attn2.to_v.weight", "transformer_blocks.8.attn2.to_out.0.weight", "transformer_blocks.8.norm3.linear_1.weight", "transformer_blocks.8.norm3.linear_2.weight", "transformer_blocks.8.ff.net.0.proj.weight", "transformer_blocks.8.ff.net.2.weight", "transformer_blocks.9.norm1.linear_1.weight", "transformer_blocks.9.norm1.linear_2.weight", "transformer_blocks.9.attn1.norm_q.weight", "transformer_blocks.9.attn1.norm_k.weight", "transformer_blocks.9.attn1.to_q.weight", "transformer_blocks.9.attn1.to_k.weight", "transformer_blocks.9.attn1.to_v.weight", "transformer_blocks.9.attn1.to_out.0.weight", "transformer_blocks.9.norm2.linear_1.weight", "transformer_blocks.9.norm2.linear_2.weight", "transformer_blocks.9.attn2.norm_q.weight", "transformer_blocks.9.attn2.norm_k.weight", "transformer_blocks.9.attn2.to_q.weight", "transformer_blocks.9.attn2.to_k.weight", "transformer_blocks.9.attn2.to_v.weight", "transformer_blocks.9.attn2.to_out.0.weight", "transformer_blocks.9.norm3.linear_1.weight", "transformer_blocks.9.norm3.linear_2.weight", "transformer_blocks.9.ff.net.0.proj.weight", "transformer_blocks.9.ff.net.2.weight", "transformer_blocks.10.norm1.linear_1.weight", "transformer_blocks.10.norm1.linear_2.weight", "transformer_blocks.10.attn1.norm_q.weight", "transformer_blocks.10.attn1.norm_k.weight", "transformer_blocks.10.attn1.to_q.weight", "transformer_blocks.10.attn1.to_k.weight", "transformer_blocks.10.attn1.to_v.weight", "transformer_blocks.10.attn1.to_out.0.weight", "transformer_blocks.10.norm2.linear_1.weight", "transformer_blocks.10.norm2.linear_2.weight", "transformer_blocks.10.attn2.norm_q.weight", "transformer_blocks.10.attn2.norm_k.weight", "transformer_blocks.10.attn2.to_q.weight", "transformer_blocks.10.attn2.to_k.weight", "transformer_blocks.10.attn2.to_v.weight", "transformer_blocks.10.attn2.to_out.0.weight", "transformer_blocks.10.norm3.linear_1.weight", "transformer_blocks.10.norm3.linear_2.weight", "transformer_blocks.10.ff.net.0.proj.weight", "transformer_blocks.10.ff.net.2.weight", "transformer_blocks.11.norm1.linear_1.weight", "transformer_blocks.11.norm1.linear_2.weight", "transformer_blocks.11.attn1.norm_q.weight", "transformer_blocks.11.attn1.norm_k.weight", "transformer_blocks.11.attn1.to_q.weight", "transformer_blocks.11.attn1.to_k.weight", "transformer_blocks.11.attn1.to_v.weight", "transformer_blocks.11.attn1.to_out.0.weight", "transformer_blocks.11.norm2.linear_1.weight", "transformer_blocks.11.norm2.linear_2.weight", "transformer_blocks.11.attn2.norm_q.weight", "transformer_blocks.11.attn2.norm_k.weight", "transformer_blocks.11.attn2.to_q.weight", "transformer_blocks.11.attn2.to_k.weight", "transformer_blocks.11.attn2.to_v.weight", "transformer_blocks.11.attn2.to_out.0.weight", "transformer_blocks.11.norm3.linear_1.weight", "transformer_blocks.11.norm3.linear_2.weight", "transformer_blocks.11.ff.net.0.proj.weight", "transformer_blocks.11.ff.net.2.weight", "transformer_blocks.12.norm1.linear_1.weight", "transformer_blocks.12.norm1.linear_2.weight", "transformer_blocks.12.attn1.norm_q.weight", "transformer_blocks.12.attn1.norm_k.weight", "transformer_blocks.12.attn1.to_q.weight", "transformer_blocks.12.attn1.to_k.weight", "transformer_blocks.12.attn1.to_v.weight", "transformer_blocks.12.attn1.to_out.0.weight", "transformer_blocks.12.norm2.linear_1.weight", "transformer_blocks.12.norm2.linear_2.weight", "transformer_blocks.12.attn2.norm_q.weight", "transformer_blocks.12.attn2.norm_k.weight", "transformer_blocks.12.attn2.to_q.weight", "transformer_blocks.12.attn2.to_k.weight", "transformer_blocks.12.attn2.to_v.weight", "transformer_blocks.12.attn2.to_out.0.weight", "transformer_blocks.12.norm3.linear_1.weight", "transformer_blocks.12.norm3.linear_2.weight", "transformer_blocks.12.ff.net.0.proj.weight", "transformer_blocks.12.ff.net.2.weight", "transformer_blocks.13.norm1.linear_1.weight", "transformer_blocks.13.norm1.linear_2.weight", "transformer_blocks.13.attn1.norm_q.weight", "transformer_blocks.13.attn1.norm_k.weight", "transformer_blocks.13.attn1.to_q.weight", "transformer_blocks.13.attn1.to_k.weight", "transformer_blocks.13.attn1.to_v.weight", "transformer_blocks.13.attn1.to_out.0.weight", "transformer_blocks.13.norm2.linear_1.weight", "transformer_blocks.13.norm2.linear_2.weight", "transformer_blocks.13.attn2.norm_q.weight", "transformer_blocks.13.attn2.norm_k.weight", "transformer_blocks.13.attn2.to_q.weight", "transformer_blocks.13.attn2.to_k.weight", "transformer_blocks.13.attn2.to_v.weight", "transformer_blocks.13.attn2.to_out.0.weight", "transformer_blocks.13.norm3.linear_1.weight", "transformer_blocks.13.norm3.linear_2.weight", "transformer_blocks.13.ff.net.0.proj.weight", "transformer_blocks.13.ff.net.2.weight", "transformer_blocks.14.norm1.linear_1.weight", "transformer_blocks.14.norm1.linear_2.weight", "transformer_blocks.14.attn1.norm_q.weight", "transformer_blocks.14.attn1.norm_k.weight", "transformer_blocks.14.attn1.to_q.weight", "transformer_blocks.14.attn1.to_k.weight", "transformer_blocks.14.attn1.to_v.weight", "transformer_blocks.14.attn1.to_out.0.weight", "transformer_blocks.14.norm2.linear_1.weight", "transformer_blocks.14.norm2.linear_2.weight", "transformer_blocks.14.attn2.norm_q.weight", "transformer_blocks.14.attn2.norm_k.weight", "transformer_blocks.14.attn2.to_q.weight", "transformer_blocks.14.attn2.to_k.weight", "transformer_blocks.14.attn2.to_v.weight", "transformer_blocks.14.attn2.to_out.0.weight", "transformer_blocks.14.norm3.linear_1.weight", "transformer_blocks.14.norm3.linear_2.weight", "transformer_blocks.14.ff.net.0.proj.weight", "transformer_blocks.14.ff.net.2.weight", "transformer_blocks.15.norm1.linear_1.weight", "transformer_blocks.15.norm1.linear_2.weight", "transformer_blocks.15.attn1.norm_q.weight", "transformer_blocks.15.attn1.norm_k.weight", "transformer_blocks.15.attn1.to_q.weight", "transformer_blocks.15.attn1.to_k.weight", "transformer_blocks.15.attn1.to_v.weight", "transformer_blocks.15.attn1.to_out.0.weight", "transformer_blocks.15.norm2.linear_1.weight", "transformer_blocks.15.norm2.linear_2.weight", "transformer_blocks.15.attn2.norm_q.weight", "transformer_blocks.15.attn2.norm_k.weight", "transformer_blocks.15.attn2.to_q.weight", "transformer_blocks.15.attn2.to_k.weight", "transformer_blocks.15.attn2.to_v.weight", "transformer_blocks.15.attn2.to_out.0.weight", "transformer_blocks.15.norm3.linear_1.weight", "transformer_blocks.15.norm3.linear_2.weight", "transformer_blocks.15.ff.net.0.proj.weight", "transformer_blocks.15.ff.net.2.weight", "transformer_blocks.16.norm1.linear_1.weight", "transformer_blocks.16.norm1.linear_2.weight", "transformer_blocks.16.attn1.norm_q.weight", "transformer_blocks.16.attn1.norm_k.weight", "transformer_blocks.16.attn1.to_q.weight", "transformer_blocks.16.attn1.to_k.weight", "transformer_blocks.16.attn1.to_v.weight", "transformer_blocks.16.attn1.to_out.0.weight", "transformer_blocks.16.norm2.linear_1.weight", "transformer_blocks.16.norm2.linear_2.weight", "transformer_blocks.16.attn2.norm_q.weight", "transformer_blocks.16.attn2.norm_k.weight", "transformer_blocks.16.attn2.to_q.weight", "transformer_blocks.16.attn2.to_k.weight", "transformer_blocks.16.attn2.to_v.weight", "transformer_blocks.16.attn2.to_out.0.weight", "transformer_blocks.16.norm3.linear_1.weight", "transformer_blocks.16.norm3.linear_2.weight", "transformer_blocks.16.ff.net.0.proj.weight", "transformer_blocks.16.ff.net.2.weight", "transformer_blocks.17.norm1.linear_1.weight", "transformer_blocks.17.norm1.linear_2.weight", "transformer_blocks.17.attn1.norm_q.weight", "transformer_blocks.17.attn1.norm_k.weight", "transformer_blocks.17.attn1.to_q.weight", "transformer_blocks.17.attn1.to_k.weight", "transformer_blocks.17.attn1.to_v.weight", "transformer_blocks.17.attn1.to_out.0.weight", "transformer_blocks.17.norm2.linear_1.weight", "transformer_blocks.17.norm2.linear_2.weight", "transformer_blocks.17.attn2.norm_q.weight", "transformer_blocks.17.attn2.norm_k.weight", "transformer_blocks.17.attn2.to_q.weight", "transformer_blocks.17.attn2.to_k.weight", "transformer_blocks.17.attn2.to_v.weight", "transformer_blocks.17.attn2.to_out.0.weight", "transformer_blocks.17.norm3.linear_1.weight", "transformer_blocks.17.norm3.linear_2.weight", "transformer_blocks.17.ff.net.0.proj.weight", "transformer_blocks.17.ff.net.2.weight", "transformer_blocks.18.norm1.linear_1.weight", "transformer_blocks.18.norm1.linear_2.weight", "transformer_blocks.18.attn1.norm_q.weight", "transformer_blocks.18.attn1.norm_k.weight", "transformer_blocks.18.attn1.to_q.weight", "transformer_blocks.18.attn1.to_k.weight", "transformer_blocks.18.attn1.to_v.weight", "transformer_blocks.18.attn1.to_out.0.weight", "transformer_blocks.18.norm2.linear_1.weight", "transformer_blocks.18.norm2.linear_2.weight", "transformer_blocks.18.attn2.norm_q.weight", "transformer_blocks.18.attn2.norm_k.weight", "transformer_blocks.18.attn2.to_q.weight", "transformer_blocks.18.attn2.to_k.weight", "transformer_blocks.18.attn2.to_v.weight", "transformer_blocks.18.attn2.to_out.0.weight", "transformer_blocks.18.norm3.linear_1.weight", "transformer_blocks.18.norm3.linear_2.weight", "transformer_blocks.18.ff.net.0.proj.weight", "transformer_blocks.18.ff.net.2.weight", "transformer_blocks.19.norm1.linear_1.weight", "transformer_blocks.19.norm1.linear_2.weight", "transformer_blocks.19.attn1.norm_q.weight", "transformer_blocks.19.attn1.norm_k.weight", "transformer_blocks.19.attn1.to_q.weight", "transformer_blocks.19.attn1.to_k.weight", "transformer_blocks.19.attn1.to_v.weight", "transformer_blocks.19.attn1.to_out.0.weight", "transformer_blocks.19.norm2.linear_1.weight", "transformer_blocks.19.norm2.linear_2.weight", "transformer_blocks.19.attn2.norm_q.weight", "transformer_blocks.19.attn2.norm_k.weight", "transformer_blocks.19.attn2.to_q.weight", "transformer_blocks.19.attn2.to_k.weight", "transformer_blocks.19.attn2.to_v.weight", "transformer_blocks.19.attn2.to_out.0.weight", "transformer_blocks.19.norm3.linear_1.weight", "transformer_blocks.19.norm3.linear_2.weight", "transformer_blocks.19.ff.net.0.proj.weight", "transformer_blocks.19.ff.net.2.weight", "transformer_blocks.20.norm1.linear_1.weight", "transformer_blocks.20.norm1.linear_2.weight", "transformer_blocks.20.attn1.norm_q.weight", "transformer_blocks.20.attn1.norm_k.weight", "transformer_blocks.20.attn1.to_q.weight", "transformer_blocks.20.attn1.to_k.weight", "transformer_blocks.20.attn1.to_v.weight", "transformer_blocks.20.attn1.to_out.0.weight", "transformer_blocks.20.norm2.linear_1.weight", "transformer_blocks.20.norm2.linear_2.weight", "transformer_blocks.20.attn2.norm_q.weight", "transformer_blocks.20.attn2.norm_k.weight", "transformer_blocks.20.attn2.to_q.weight", "transformer_blocks.20.attn2.to_k.weight", "transformer_blocks.20.attn2.to_v.weight", "transformer_blocks.20.attn2.to_out.0.weight", "transformer_blocks.20.norm3.linear_1.weight", "transformer_blocks.20.norm3.linear_2.weight", "transformer_blocks.20.ff.net.0.proj.weight", "transformer_blocks.20.ff.net.2.weight", "transformer_blocks.21.norm1.linear_1.weight", "transformer_blocks.21.norm1.linear_2.weight", "transformer_blocks.21.attn1.norm_q.weight", "transformer_blocks.21.attn1.norm_k.weight", "transformer_blocks.21.attn1.to_q.weight", "transformer_blocks.21.attn1.to_k.weight", "transformer_blocks.21.attn1.to_v.weight", "transformer_blocks.21.attn1.to_out.0.weight", "transformer_blocks.21.norm2.linear_1.weight", "transformer_blocks.21.norm2.linear_2.weight", "transformer_blocks.21.attn2.norm_q.weight", "transformer_blocks.21.attn2.norm_k.weight", "transformer_blocks.21.attn2.to_q.weight", "transformer_blocks.21.attn2.to_k.weight", "transformer_blocks.21.attn2.to_v.weight", "transformer_blocks.21.attn2.to_out.0.weight", "transformer_blocks.21.norm3.linear_1.weight", "transformer_blocks.21.norm3.linear_2.weight", "transformer_blocks.21.ff.net.0.proj.weight", "transformer_blocks.21.ff.net.2.weight", "transformer_blocks.22.norm1.linear_1.weight", "transformer_blocks.22.norm1.linear_2.weight", "transformer_blocks.22.attn1.norm_q.weight", "transformer_blocks.22.attn1.norm_k.weight", "transformer_blocks.22.attn1.to_q.weight", "transformer_blocks.22.attn1.to_k.weight", "transformer_blocks.22.attn1.to_v.weight", "transformer_blocks.22.attn1.to_out.0.weight", "transformer_blocks.22.norm2.linear_1.weight", "transformer_blocks.22.norm2.linear_2.weight", "transformer_blocks.22.attn2.norm_q.weight", "transformer_blocks.22.attn2.norm_k.weight", "transformer_blocks.22.attn2.to_q.weight", "transformer_blocks.22.attn2.to_k.weight", "transformer_blocks.22.attn2.to_v.weight", "transformer_blocks.22.attn2.to_out.0.weight", "transformer_blocks.22.norm3.linear_1.weight", "transformer_blocks.22.norm3.linear_2.weight", "transformer_blocks.22.ff.net.0.proj.weight", "transformer_blocks.22.ff.net.2.weight", "transformer_blocks.23.norm1.linear_1.weight", "transformer_blocks.23.norm1.linear_2.weight", "transformer_blocks.23.attn1.norm_q.weight", "transformer_blocks.23.attn1.norm_k.weight", "transformer_blocks.23.attn1.to_q.weight", "transformer_blocks.23.attn1.to_k.weight", "transformer_blocks.23.attn1.to_v.weight", "transformer_blocks.23.attn1.to_out.0.weight", "transformer_blocks.23.norm2.linear_1.weight", "transformer_blocks.23.norm2.linear_2.weight", "transformer_blocks.23.attn2.norm_q.weight", "transformer_blocks.23.attn2.norm_k.weight", "transformer_blocks.23.attn2.to_q.weight", "transformer_blocks.23.attn2.to_k.weight", "transformer_blocks.23.attn2.to_v.weight", "transformer_blocks.23.attn2.to_out.0.weight", "transformer_blocks.23.norm3.linear_1.weight", "transformer_blocks.23.norm3.linear_2.weight", "transformer_blocks.23.ff.net.0.proj.weight", "transformer_blocks.23.ff.net.2.weight", "transformer_blocks.24.norm1.linear_1.weight", "transformer_blocks.24.norm1.linear_2.weight", "transformer_blocks.24.attn1.norm_q.weight", "transformer_blocks.24.attn1.norm_k.weight", "transformer_blocks.24.attn1.to_q.weight", "transformer_blocks.24.attn1.to_k.weight", "transformer_blocks.24.attn1.to_v.weight", "transformer_blocks.24.attn1.to_out.0.weight", "transformer_blocks.24.norm2.linear_1.weight", "transformer_blocks.24.norm2.linear_2.weight", "transformer_blocks.24.attn2.norm_q.weight", "transformer_blocks.24.attn2.norm_k.weight", "transformer_blocks.24.attn2.to_q.weight", "transformer_blocks.24.attn2.to_k.weight", "transformer_blocks.24.attn2.to_v.weight", "transformer_blocks.24.attn2.to_out.0.weight", "transformer_blocks.24.norm3.linear_1.weight", "transformer_blocks.24.norm3.linear_2.weight", "transformer_blocks.24.ff.net.0.proj.weight", "transformer_blocks.24.ff.net.2.weight", "transformer_blocks.25.norm1.linear_1.weight", "transformer_blocks.25.norm1.linear_2.weight", "transformer_blocks.25.attn1.norm_q.weight", "transformer_blocks.25.attn1.norm_k.weight", "transformer_blocks.25.attn1.to_q.weight", "transformer_blocks.25.attn1.to_k.weight", "transformer_blocks.25.attn1.to_v.weight", "transformer_blocks.25.attn1.to_out.0.weight", "transformer_blocks.25.norm2.linear_1.weight", "transformer_blocks.25.norm2.linear_2.weight", "transformer_blocks.25.attn2.norm_q.weight", "transformer_blocks.25.attn2.norm_k.weight", "transformer_blocks.25.attn2.to_q.weight", "transformer_blocks.25.attn2.to_k.weight", "transformer_blocks.25.attn2.to_v.weight", "transformer_blocks.25.attn2.to_out.0.weight", "transformer_blocks.25.norm3.linear_1.weight", "transformer_blocks.25.norm3.linear_2.weight", "transformer_blocks.25.ff.net.0.proj.weight", "transformer_blocks.25.ff.net.2.weight", "transformer_blocks.26.norm1.linear_1.weight", "transformer_blocks.26.norm1.linear_2.weight", "transformer_blocks.26.attn1.norm_q.weight", "transformer_blocks.26.attn1.norm_k.weight", "transformer_blocks.26.attn1.to_q.weight", "transformer_blocks.26.attn1.to_k.weight", "transformer_blocks.26.attn1.to_v.weight", "transformer_blocks.26.attn1.to_out.0.weight", "transformer_blocks.26.norm2.linear_1.weight", "transformer_blocks.26.norm2.linear_2.weight", "transformer_blocks.26.attn2.norm_q.weight", "transformer_blocks.26.attn2.norm_k.weight", "transformer_blocks.26.attn2.to_q.weight", "transformer_blocks.26.attn2.to_k.weight", "transformer_blocks.26.attn2.to_v.weight", "transformer_blocks.26.attn2.to_out.0.weight", "transformer_blocks.26.norm3.linear_1.weight", "transformer_blocks.26.norm3.linear_2.weight", "transformer_blocks.26.ff.net.0.proj.weight", "transformer_blocks.26.ff.net.2.weight", "transformer_blocks.27.norm1.linear_1.weight", "transformer_blocks.27.norm1.linear_2.weight", "transformer_blocks.27.attn1.norm_q.weight", "transformer_blocks.27.attn1.norm_k.weight", "transformer_blocks.27.attn1.to_q.weight", "transformer_blocks.27.attn1.to_k.weight", "transformer_blocks.27.attn1.to_v.weight", "transformer_blocks.27.attn1.to_out.0.weight", "transformer_blocks.27.norm2.linear_1.weight", "transformer_blocks.27.norm2.linear_2.weight", "transformer_blocks.27.attn2.norm_q.weight", "transformer_blocks.27.attn2.norm_k.weight", "transformer_blocks.27.attn2.to_q.weight", "transformer_blocks.27.attn2.to_k.weight", "transformer_blocks.27.attn2.to_v.weight", "transformer_blocks.27.attn2.to_out.0.weight", "transformer_blocks.27.norm3.linear_1.weight", "transformer_blocks.27.norm3.linear_2.weight", "transformer_blocks.27.ff.net.0.proj.weight", "transformer_blocks.27.ff.net.2.weight".
Unexpected key(s) in state_dict: "patch_embed2.proj.weight", "input_hint_block.0.weight", "input_hint_block.0.bias", "input_hint_block.2.weight", "input_hint_block.2.bias", "input_hint_block.4.weight", "input_hint_block.4.bias", "input_hint_block.6.weight", "input_hint_block.6.bias", "input_hint_block.8.weight", "input_hint_block.8.bias", "input_hint_block.10.weight", "input_hint_block.10.bias", "input_hint_block.12.weight", "input_hint_block.12.bias", "input_hint_block.14.weight", "input_hint_block.14.bias", "transformer_blocks.0zero_blocks.block0.weight", "transformer_blocks.0zero_blocks.block0.bias", "transformer_blocks.1zero_blocks.block1.weight", "transformer_blocks.1zero_blocks.block1.bias", "transformer_blocks.2zero_blocks.block2.weight", "transformer_blocks.2zero_blocks.block2.bias".
what is correct way to run "convert_cosmos_to_diffusers.py" ?
please help me!
Reproduction
PYTHONPATH=. python ../convert_cosmos_to_diffusers.py --transformer_type Cosmos-1.0-Diffusion-7B-Video2World --vae_type CV8x8x8-1.0 --transformer_ckpt_path ../fsdp_edge_v1/iter_000016000_ema_model_only.pt --output_path ./convert_to_diffusers
Logs
System Info
python 3.10
H100
Who can help?
No response