Waifu DiffusionとTrinart Stable DiffusionモデルをStable Diffusion web UI(AUTOMATIC1111) Google Colab で動かす

2022年11月19日2022年11月23日

Colaboratory「ノートブック」作成

下記 URL で StableDiffusionUI-Voldemort.ipynbを開いて名前を付けて保存します

https://colab.research.google.com/drive/1Iy-xW9t1-OQWhb0hNxueGij8phCyluOh

Download the model from huggingface. モデルデータダウンロードのところを書き換えます

Stable diffusion

model_url = “https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt" #@param {type:"string"}
model_filename = “sd-v1-4.ckpt" #@param {type:"string"}
user_token = “" #@param {type:"string"}
user_header = f"\"Authorization: Bearer {user_token}\""
!wget –header={user_header} {model_url} -O models/{model_filename}

Waifu Diffusion

model_url2 = “https://huggingface.co/hakurei/waifu-diffusion-v1-3/resolve/main/wd-v1-3-float16.ckpt" #@param {type:"string"}
model_filename2 ="wd-v1-3-float16.ckpt" #@param {type:"string"}
!wget {model_url2} -O models/{model_filename2}

Trinart Stable Diffusion

model_url3 = “https://huggingface.co/naclbit/trinart_stable_diffusion_v2/resolve/main/trinart2_step115000.ckpt" #@param {type:"string"}
model_filename3 ="trinart_characters_it4_v1.ckpt" #@param {type:"string"}
!wget {model_url3} -O models/{model_filename3}

個別に置き換えてもいいですが 3個並べて一度にダウンロードするととWeb の実行画面で切り替えができます