Create Dockerfile

Dockerfile

@@ -0,0 +1,40 @@
+FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
+
+# 必要なパッケージをインストール
+RUN apt-get update && apt-get install -y \
+    git wget python3 python3-pip && \
+    apt-get clean && rm -rf /var/lib/apt/lists/*
+
+# Pythonのデフォルト実行ファイルをpython3に変更
+RUN ln -s /usr/bin/python3 /usr/bin/python
+
+# Hugging Face APIトークンを環境変数として設定
+ARG HF_TOKEN
+ENV HF_TOKEN=${HF_TOKEN}
+
+# リポジトリをクローン
+WORKDIR /
+RUN git clone https://github.com/lllyasviel/stable-diffusion-webui-forge.git
+
+# 必要なモデルをダウンロード
+WORKDIR /stable-diffusion-webui-forge/models/text_encoder
+RUN wget --header="Authorization: Bearer ${HF_TOKEN}" \
+    https://huggingface.co/black-forest-labs/FLUX.1-dev/resolve/main/ae.safetensors \
+    https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/text_encoder/model.fp16.safetensors \
+    https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/text_encoder/model.safetensors \
+    https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/clip_l.safetensors
+
+RUN mkdir -p /stable-diffusion-webui-forge/models/clip/
+WORKDIR /stable-diffusion-webui-forge/models/clip/
+RUN wget --header="Authorization: Bearer ${HF_TOKEN}" \
+    "https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/clip_l.safetensors" \
+    "https://huggingface.co/openai/clip-vit-large-patch14/resolve/main/model.safetensors"
+
+WORKDIR /stable-diffusion-webui-forge/models/Stable-diffusion
+RUN wget --header="Authorization: Bearer ${HF_TOKEN}" \
+    https://huggingface.co/stabilityai/stable-diffusion-3.5-large/resolve/main/sd3.5_large.safetensors \
+    https://huggingface.co/lllyasviel/flux1-dev-bnb-nf4/resolve/main/flux1-dev-bnb-nf4-v2.safetensors
+
+# WebUIの起動
+WORKDIR /stable-diffusion-webui-forge
+CMD ["python3", "launch.py", "--share"]