RuntimeError: mat1 and mat2 shapes cannot be multiplied (2050x768 and 1536x1536)

[YuyangYin]

My command is 'python flow/scripts/infer.py --input assets/images/cyan_car.png'. The problem is as followed. How can I fix that? Thanks

Number of parameters in VAE: 340.30M Loaded VAE from pretrained/vae.pt Number of parameters in DiT: 1249.51M Loading weights from pretrained/flow.pt Output directory: output/flow_big_parts_strict_pvae_20250813_155828 Processing cyan_car Flow Sampling: 0%| | 0/50 [00:00<?, ?it/s] Traceback (most recent call last): File "/vepfs-cnbj54effbf51f70/yyy/PartPacker/flow/scripts/infer.py", line 128, in <module> results = model(data, num_steps=args.num_steps, cfg_scale=args.cfg_scale) File "/vepfs-cnbj54effbf51f70/conda/anaconda3/envs/partpacker/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "/vepfs-cnbj54effbf51f70/conda/anaconda3/envs/partpacker/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl return forward_call(*args, **kwargs) File "/vepfs-cnbj54effbf51f70/conda/anaconda3/envs/partpacker/lib/python3.10/site-packages/torch/utils/_contextlib.py", line 116, in decorate_context return func(*args, **kwargs) File "/vepfs-cnbj54effbf51f70/yyy/PartPacker/vae/utils.py", line 66, in wrapper result = func(*args, **kwargs) File "/vepfs-cnbj54effbf51f70/yyy/PartPacker/flow/model.py", line 325, in forward pred = self.dit(x_input, cond_input, timesteps).float() File "/vepfs-cnbj54effbf51f70/conda/anaconda3/envs/partpacker/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "/vepfs-cnbj54effbf51f70/conda/anaconda3/envs/partpacker/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl return forward_call(*args, **kwargs) File "/vepfs-cnbj54effbf51f70/yyy/PartPacker/flow/modules/dit.py", line 229, in forward x = layer(x, c, t_emb) File "/vepfs-cnbj54effbf51f70/conda/anaconda3/envs/partpacker/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "/vepfs-cnbj54effbf51f70/conda/anaconda3/envs/partpacker/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl return forward_call(*args, **kwargs) File "/vepfs-cnbj54effbf51f70/yyy/PartPacker/flow/modules/dit.py", line 98, in forward return self._forward(x, c, t_emb) File "/vepfs-cnbj54effbf51f70/yyy/PartPacker/flow/modules/dit.py", line 115, in _forward x = x + self.attn2(h, c) File "/vepfs-cnbj54effbf51f70/conda/anaconda3/envs/partpacker/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "/vepfs-cnbj54effbf51f70/conda/anaconda3/envs/partpacker/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl return forward_call(*args, **kwargs) File "/vepfs-cnbj54effbf51f70/yyy/PartPacker/vae/modules/attention.py", line 251, in forward k = self.k_proj(context) File "/vepfs-cnbj54effbf51f70/conda/anaconda3/envs/partpacker/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "/vepfs-cnbj54effbf51f70/conda/anaconda3/envs/partpacker/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl return forward_call(*args, **kwargs) File "/vepfs-cnbj54effbf51f70/conda/anaconda3/envs/partpacker/lib/python3.10/site-packages/torch/nn/modules/linear.py", line 125, in forward return F.linear(input, self.weight, self.bias) RuntimeError: mat1 and mat2 shapes cannot be multiplied (2050x768 and 1536x1536)