Windows上的Dreambooth 可以采用ShivamShrirao的优化版本来节省显存,
diffusers/examples/dreambooth at main · ShivamShrirao/diffusers · GitHub
但是,由于相关链接库的原因,用于在Linux上的部署方法无法直接在windows上使用,由于同样的原因,该优化版本在colab上最低9.9G的显存需求在windows上应该稍高,因此推荐至少使用显存12G的显卡。笔者仅在16G显存PC上测试成功,12G理论可以,待测试。
修改或覆盖原始库中的文件前请备份!
准备环境 Git
,Python
,MiniConda
(或 MiniConda
)。