引导扩散
这是的代码库。
该存储库基于 ,并针对分类器调节和架构改进进行了修改。
下载预训练模型
我们已经发布了论文中主要模型的检查点。 在使用这些型号之前,请查看相应的以了解这些型号的预期用途和限制。
以下是每个模型检查点的下载链接:
64x64 分类器:
64x64 扩散:
128x128 分类器:
128x128 扩散:
256x256 分类器:
256x256 扩散:
256x256 扩散(非类条件):
512x512 分类器:
512x512 扩散:
64x64 -> 256x256 上采样器:
128x128 -> 512x512 上采样器:
LSUN 卧室: lsun_bedroom.pt
LSUN猫: lsun_cat.pt
LSUN 马: lsun_horse.pt
LSUN 马(无辍学): lsun_horse_nodropout.pt
2023-03-05 10:26:29
54KB
Python
1