Weighting and Reasoning

sail-sg / MDT

Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)

Apache License 2.0

500 stars 35 forks source link

Closed liwenyang-911 closed 5 months ago

liwenyang-911 commented 5 months ago

If I want to use the weights obtained by MDTv2_B_2 model training for inference, what should I modify in file infer_mdt.py?

gasvn commented 5 months ago

You can change these: