DiT 在 1.1 版本推理出错

[songh11]

报错信息：
RuntimeError: Error(s) in loading state_dict for DiT:
size mismatch for x_embedder.proj.weight: copying a param with shape torch.Size([1152, 4, 2, 2]) from checkpoint, the shape in current model is torch.Size([1152, 4, 1, 2, 2]).

[github-actions]

This issue is stale because it has been open for 7 days with no activity.

[zhengzangw]

We do not reproduce this error. You should use automatic downloading for DiT weight. Otherwise, you need manually do something like:

Open-Sora/opensora/utils/ckpt_utils.py

Line 39 in c6cc021

if name in ["DiT-XL-2-512x512.pt", "DiT-XL-2-256x256.pt"]:

[github-actions]

This issue is stale because it has been open for 7 days with no activity.