darkstorm2150 commited on
Commit
78ef80b
·
verified ·
1 Parent(s): 7126efe

Update Dockerfile

Browse files

added A10G xformer and testing model issue with loading on the proper folder.

Files changed (1) hide show
  1. Dockerfile +5 -3
Dockerfile CHANGED
@@ -11,9 +11,11 @@ USER user
11
 
12
  # Pre-download model in a separate stage for better caching
13
  FROM base AS model-downloader
14
- ADD https://huggingface.co/darkstorm2150/OpenGen/resolve/main/OpenGen%20v1.0.safetensors /tmp/model.safetensors
 
15
  USER root
16
- RUN mv /tmp/model.safetensors /content/stable-diffusion-webui/models/Stable-diffusion/
 
17
  USER user
18
 
19
  # Main build stage with optimized layers
@@ -33,7 +35,7 @@ RUN pip3 install --upgrade pip==23.3.1 && \
33
  numexpr \
34
  httpx==0.24.1 \
35
  pytorch_lightning==1.7.6 \
36
- 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 \
37
  --pre triton
38
 
39
  # Clone repository and apply patches in a single step
 
11
 
12
  # Pre-download model in a separate stage for better caching
13
  FROM base AS model-downloader
14
+ ADD https://huggingface.co/darkstorm2150/OpenGen/resolve/main/OpenGen%20v1.0.safetensors \
15
+ /tmp/OpenGen%20v1.0.safetensors
16
  USER root
17
+ RUN mv "/tmp/OpenGen%20v1.0.safetensors" \
18
+ "/content/stable-diffusion-webui/models/Stable-diffusion/OpenGen v1.0.safetensors"
19
  USER user
20
 
21
  # Main build stage with optimized layers
 
35
  numexpr \
36
  httpx==0.24.1 \
37
  pytorch_lightning==1.7.6 \
38
+ 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 \
39
  --pre triton
40
 
41
  # Clone repository and apply patches in a single step