Mismatch in last layers during inference

[Trinita21typeset]

I am running the inference and this is what I am getting. The command that I ran: python sample.py -m "DiT-XL/2" --text "a person is walking on the street" --ckpt /home/nlp/open_sora/Open-Sora/pretrained_models/DiT-XL-2-256x256.pt --height 256 --width 256 --fps 10 --sec 5 --disable-cfg I downloaded the checkpoints using download.py in the given repo.

[ver217]

Hi, as we modified the modeling, you cannot load pretrained DiT weights directly

[Trinita21typeset]

Then can you let me know how to load the weights or where do I download the weights so that I can run the model locally?

[zhengzangw]

We have updated our code and inference. Please see here for more instructions.