diffusersモデルをckptに変換してみる

diffusersモデルをckptに変換して、AUTOMATIC1111でも使えるようにしてみます。

1. diffusersモデルの準備

「Stable Diffusion Dreambooth Concepts Library」でDreamBoothのファインチューニングした時の出力も、diffusersモデルです。
dreambooth-conceptフォルダに出力されています。

2. diffusersモデルをckptに変換

diffusersモデルをckptに変換するコードが以下で公開されてるので、使わせてもらいます。

・convert_diffusers_to_sd.py

jachiam’s gists

「Google Colab」での変換手順は、次のとおりです。

(1) convert_diffusers_to_sd.pyのダウンロード。

!git clone https://gist.github.com/jachiam/8a5c0b607e38fcc585168b90c686eb05 jachiam

(2) diffusersモデルをカレントフォルダに配置。
(3) convert_diffusers_to_sd.pyの実行。
diffusersモデル「dreambooth-concept」をckpt「cat.ckpt」(4GB)に変換しています。

!python jachiam/convert_diffusers_to_sd.py --model_path dreambooth-concept --checkpoint_path cat.ckpt

-h, --help : ヘルプ
--model_path <MODEL_PATH> : 変換するdiffusersモデルのパス
--checkpoint_path <CHECKPOINT_PATH> : 出力するckptのパス
--half : 重みを半精度で保存

ckptに変換できたので、AUTOMATIC1111でも利用できるようになりました。

Unityではじめる ROS・人工知能 ロボットプログラミング実践入門