How to Use Saved Checkpoints to Generate Images?

Tencent / HunyuanDiT

Hunyuan-DiT : A Powerful Multi-Resolution Diffusion Transformer with Fine-Grained Chinese Understanding

https://dit.hunyuan.tencent.com/

Other

2.59k stars 180 forks source link

How to Use Saved Checkpoints to Generate Images? #116

Open iamwangyabin opened 6 days ago

iamwangyabin commented 6 days ago

After fully fine-tuning, I can get the following weights from the log directory: mp_rank_00_model_states.pt zero_pp_rank_0_mp_rank_00_optim_states.pt How should I use these weights?