darkstorm2150 commited on
Commit
043735b
1 Parent(s): 983ac98

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -4
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- # Dockerfile Public T4
2
 
3
  # https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/11.7.1/ubuntu2204/devel/cudnn8/Dockerfile
4
  # FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04
@@ -15,7 +15,7 @@ USER user
15
 
16
  RUN pip3 install --upgrade pip
17
  RUN pip install torchmetrics==0.11.4
18
- RUN pip install https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.16/xformers-0.0.16+814314d.d20230118-cp310-cp310-linux_x86_64.whl
19
  RUN pip install --pre triton
20
  RUN pip install numexpr
21
 
@@ -45,8 +45,7 @@ RUN rm -rfv /content/stable-diffusion-webui/scripts/
45
  ADD --chown=user https://github.com/camenduru/webui-docker/raw/main/shared-config.json /content/shared-config.json
46
  ADD --chown=user https://github.com/camenduru/webui-docker/raw/main/shared-ui-config.json /content/shared-ui-config.json
47
 
48
- ADD --chown=user https://huggingface.co/darkstorm2150/Protogen_x3.4_Official_Release/resolve/main/ProtoGen_X3.4.safetensors /content/stable-diffusion-webui/models/Stable-diffusion/ProtoGen_X3.4.safetensors
49
-
50
 
51
  EXPOSE 7860
52
 
 
1
+ # Dockerfile Public A10G
2
 
3
  # https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/11.7.1/ubuntu2204/devel/cudnn8/Dockerfile
4
  # FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04
 
15
 
16
  RUN pip3 install --upgrade pip
17
  RUN pip install torchmetrics==0.11.4
18
+ RUN pip install https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.16/xformers-0.0.16+814314d.d20230119.A10G-cp310-cp310-linux_x86_64.whl
19
  RUN pip install --pre triton
20
  RUN pip install numexpr
21
 
 
45
  ADD --chown=user https://github.com/camenduru/webui-docker/raw/main/shared-config.json /content/shared-config.json
46
  ADD --chown=user https://github.com/camenduru/webui-docker/raw/main/shared-ui-config.json /content/shared-ui-config.json
47
 
48
+ ADD --chown=user https://huggingface.co/darkstorm2150/OpenGen/resolve/main/OpenGen%20v1.0.safetensors /content/stable-diffusion-webui/models/Stable-diffusion/OpenGen%20v1.0.safetensors
 
49
 
50
  EXPOSE 7860
51