soiz1 commited on
Commit
e103758
·
verified ·
1 Parent(s): c204b72

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -4
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
2
 
3
  # 必要なパッケージをインストール
4
  RUN apt-get update && apt-get install -y \
@@ -10,7 +10,7 @@ RUN ln -s /usr/bin/python3 /usr/bin/python
10
 
11
  # Hugging Face APIトークンを環境変数として設定
12
  ARG HF_TOKEN
13
- ENV HF_TOKEN=${HF_TOKEN}
14
 
15
  # リポジトリをクローン
16
  WORKDIR /
@@ -35,6 +35,6 @@ RUN wget --header="Authorization: Bearer ${HF_TOKEN}" \
35
  https://huggingface.co/stabilityai/stable-diffusion-3.5-large/resolve/main/sd3.5_large.safetensors \
36
  https://huggingface.co/lllyasviel/flux1-dev-bnb-nf4/resolve/main/flux1-dev-bnb-nf4-v2.safetensors
37
 
38
- # WebUIの起動
39
  WORKDIR /stable-diffusion-webui-forge
40
- CMD ["python3", "launch.py", "--share"]
 
1
+ FROM ubuntu:22.04
2
 
3
  # 必要なパッケージをインストール
4
  RUN apt-get update && apt-get install -y \
 
10
 
11
  # Hugging Face APIトークンを環境変数として設定
12
  ARG HF_TOKEN
13
+ ENV HF_TOKEN=${token}
14
 
15
  # リポジトリをクローン
16
  WORKDIR /
 
35
  https://huggingface.co/stabilityai/stable-diffusion-3.5-large/resolve/main/sd3.5_large.safetensors \
36
  https://huggingface.co/lllyasviel/flux1-dev-bnb-nf4/resolve/main/flux1-dev-bnb-nf4-v2.safetensors
37
 
38
+ # WebUIの起動(CPUモード)
39
  WORKDIR /stable-diffusion-webui-forge
40
+ CMD ["python3", "launch.py", "--share", "--disable-torch-cuda"]