search

sail-sg / MDT

Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)
Apache License 2.0
500 stars 35 forks source link

Inquiry of pretrained checkpoint #6

Closed OliverRensu closed 5 months ago

OliverRensu commented 1 year ago

Thanks for your fantastic work! May I ask when you will release the pretrained checkpoint?

gasvn commented 1 year ago

Thanks for waiting. The pretrained model is now on hugglingface Pretrained model download, you can also download it with:

from huggingface_hub import snapshot_download
models_path = snapshot_download("shgao/MDT-XL2")
ckpt_model_path = os.path.join(models_path, "mdt_xl2_v1_ckpt.pt")

A hugglingface demo is on DEMO.

alexklho commented 5 months ago

@gasvn Can you please also provide the pretrained checkpoints of MDT-S2 and MDT-B2?

gasvn commented 5 months ago

Sorry, I don't have the ckpt for s2/b2 now becasue I no longer have the access to the server for training them. The training cost for s2/b2 is relatively small.