Scalable Diffusion Models with Transformers (DiT) 在Pytorch当中的实现

---

目录

1. 仓库更新 Top News
2. 所需环境 Environment
3. 文件下载 Download
4. 预测步骤 How2predict
5. 参考资料 Reference

Top News

2024-01:创建仓库，支持简单预测，并且将DiT网络相关代码从diffusers中扒出，方便学习。

所需环境

torch==1.7.1以上

文件下载

预测所需的权重可以在百度网盘下载。
链接: https://pan.baidu.com/s/1xy_mujfbTs7gEITNQWf5Kw?pwd=kx7r
提取码: kx7r

预测步骤

a、使用预训练权重

1. 下载完库后解压，在百度网盘下载权值，放入model_data
2. 运行predict.py.

Reference

https://github.com/facebookresearch/DiT
https://github.com/huggingface/diffusers