reForgeとForge用の自作paperspaceコンテナ:v3 の機能紹介など

2024年10月27日 10:25

reForgeとForge用の環境に合わせたpaperspace用の自作コンテナです。

使用は自己責任でお願いします。

以下ので使用できます。

コンテナイメージ名

shibashiba2/paperspace-gradient-base-pytorch241:v3.0

＜機能紹介＞

基本はv2.2と同じなのですが、v2.2はforgeとreforgeに対応していませんでしたので、それらに対応したものになります。

reForgeのコマンドを参考まで記載しておきます。

①依存環境は現時点(2024/10)のものに対応
以下のコマンドを行うことでreForgeの起動ができました。
追加でpythonパッケージをインストールしなくても問題ありません。

インストールコマンド：dev stream版

%cd /tmp
!git clone --branch dev_upstream https://github.com/Panchovix/stable-diffusion-webui-reForge.git

起動コマンド

%cd /tmp/stable-diffusion-webui-reForge
!python3.10 launch.py --enable-insecure-extension-access --share --gradio-queue --xformers --pin-shared-memory --cuda-malloc --cuda-stream

②huggingfaceからのdiffusers形式のsdxlダウンロード用のスクリプトが事前に入っている（前記事のそのまま配置しています(urlなど適宜修正)）

forgeとreForgeはインストールするフォルダ名が異なるので修正する必要があります。reForgeだと以下

# モデルのURLとローカルパスのリスト
models = [
    {
        "base_url": "https://huggingface.co/John6666/cyberrealistic-pony-v63-sdxl/resolve/main",
        "base_path": "/tmp/stable-diffusion-webui-reForge/models/Stable-diffusion/cyberrealistic-pony-v63-sdxl"
    },
    {
        "base_url": "https://huggingface.co/John6666/t3-xltest2-sdxl/resolve/main",
        "base_path": "/tmp/stable-diffusion-webui-reForge/models/Stable-diffusion/t3-xltest2-sdxl"
    },
    {
        "base_url": "https://huggingface.co/John6666/wai-real-cn-v12-sdxl/resolve/main",
        "base_path": "/tmp/stable-diffusion-webui-reForge/models/Stable-diffusion/wai-real-cn-v12-sdxl"
    }
]

# 各モデルについてスクリプトを実行
for model in models:
    base_url = model["base_url"]
    base_path = model["base_path"]
    print(f"Downloading model from {base_url} to {base_path}")
    !python3 /usr/src/app/modeldownload_from_hugginface.py {base_url} {base_path}

③FLUX model downloadコマンドで、huggingfaceからDiffusers形式のFLUXモデルのunet部分がコマンドのみでダウンロード出来ます。
※これはForge用ですので、reForgeのはスクリプトの稼働を確かめるものです。

# モデルのURLとローカルパスのリスト
models = [
    {
        "base_url": "https://huggingface.co/John6666/wai-ani-flux-v10forfp8-fp8-flux/resolve/main",
        "base_path": "/tmp/stable-diffusion-webui-re/forge/models/Stable-diffusion/wai-ani-flux-v10forfp8-fp8-flux"
    }
]

# 各モデルについてスクリプトを実行
for model in models:
    base_url = model["base_url"]
    base_path = model["base_path"]
    print(f"Downloading model from {base_url} to {base_path}")
    # シェルコマンドを適切に実行
    !python3 /usr/src/app/FLUXmodeldownload_from_hugginface.py "{base_url}" "{base_path}"

④Embeddingsの使用
適当に選んだもの(pony系とSD1.5)のものを以下のコマンドで配置することが可能です。

!cp -r /usr/src/app/embeddings* /tmp/stable-diffusion-webui-reForge/embeddings

⑤Lora：SD1.5,Pony系,Fluxのもの
適当に選んだもの(pony系のLoraを以下のコマンドで配置することが可能です。

!cp -r /usr/src/app/loras* /tmp/stable-diffusion-webui-reForge/models/Lora

コントロールネットのダウンロードコマンド

# コントロールネットDEPTHfull
!cd /tmp/stable-diffusion-webui-reForge/models/ControlNet && \
wget -nc https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/diffusers_xl_depth_full.safetensors?download=true -O diffusers_xl_depth_full.safetensors

# コントロールネットip-adapter
!cd /tmp/stable-diffusion-webui-reForge/models/ControlNet && \
wget -nc https://huggingface.co/lllyasviel/sd_control_collection/resolve/main/ip-adapter_xl.pth?download=true -O ip-adapter_xl.pth

# コントロールネットmistoline
!cd /tmp/stable-diffusion-webui-reForge/models/ControlNet && \
curl --progress-bar -L -o mistoLine_fp16.safetensors https://huggingface.co/TheMistoAI/MistoLine/resolve/main/mistoLine_fp16.safetensors?download=true

# anytest
!cd /tmp/stable-diffusion-webui-reForge/models/ControlNet && \
curl --progress-bar -L -o CN-anytest_v4-marged.safetensors https://huggingface.co/2vXpSwA7/iroiro-lora/resolve/main/test_controlnet2/CN-anytest_v4-marged.safetensors?download=true