huggingface-diffusers
PyTorch 中用于圖像和音頻生成的最先進(jìn)的擴(kuò)散模型
Our library is designed with a focus on usability over performance, simple over easy,
and customizability over abstractions.
地址:https://github.com/huggingface/diffusers
huggingface-diffusers比較消耗顯卡內(nèi)存
ShivamShrirao-diffusers
Huggingface Diffusers的修改版本,經(jīng)過(guò)優(yōu)化以在較低 VRAM GPU 上獲得更好的性能。
地址:https://github.com/ShivamShrirao/diffusers
sd_dreambooth_extension
作為ShivamShrirao-diffusers的web版本,也是webui的默認(rèn)插件地址
地址:https://github.com/d8ahazard/sd_dreambooth_extension
Dreambooth-Stable-Diffusion
背景
這是 Google 的Dreambooth with Stable Diffusion的一個(gè)實(shí)現(xiàn)。最初的 Dreambooth 基于
Imagen文本到圖像模型。但是,Imagen 的模型和預(yù)訓(xùn)練權(quán)重均不可用。
為了讓人們能夠通過(guò)幾個(gè)示例微調(diào)文本到圖像模型,我在穩(wěn)定擴(kuò)散上實(shí)現(xiàn)了 Dreambooth 的想法
地址:https://github.com/XavierXiao/Dreambooth-Stable-Diffusion
fast-stable-diffusion
地址 :https://github.com/TheLastBen/fast-stable-diffusion
提供了在colab訓(xùn)練的腳本