【Consistency Models】复现指导以及效果分析。
1、先看结果
如果没看过论文的建议先看看总结笔记:【终结扩散模型】Consistency Models.OpenAI开源新模型代码,一步成图,1秒18张
用到的模型是cd_imagenet64_lpips.pt。
采样代码为:(依据显存来改batch_size的大小)
mpiexec -n 2 python image_sample.py --batch_size 64 --training_mode consistency_distillation