I have a batch of image and text pairs, and I would like to use your Large-DiT-T2I to train a Text-to-Image DiT model. Once I have installed the necessary packages, I will download consolidated_ema.00-of-01.pth and model.args.pth to the pretrained model folder, and then use the following command to start:
python -u demo.py --ckpt pretrained_model
Then I received the following error message:
May I ask how to solve this problem? I hope to receive your reply
Thank you for your amazing open source project!
I have a batch of image and text pairs, and I would like to use your Large-DiT-T2I to train a Text-to-Image DiT model. Once I have installed the necessary packages, I will download consolidated_ema.00-of-01.pth and model.args.pth to the pretrained model folder, and then use the following command to start:
python -u demo.py --ckpt pretrained_model
Then I received the following error message:
May I ask how to solve this problem? I hope to receive your reply