radames commited on
Commit
f7444e4
·
1 Parent(s): 8122c97

missing xformers

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -48,7 +48,8 @@ WORKDIR $HOME/app
48
  RUN git clone https://github.com/lllyasviel/Fooocus . && \
49
  # pin to a specific commit
50
  git checkout 18a8c6655208c31216b83bd93381cb12993597c2 && \
51
- pip install --no-cache-dir -r requirements_versions.txt --extra-index-url https://download.pytorch.org/whl/cu118
 
52
 
53
  # remove the next line if you're running on your own GPU, it set max images to 3 and disables the API
54
  RUN sed -i "s|image_number = gr.Slider(label='Image Number', minimum=1, maximum=32|image_number = gr.Slider(label='Image Number', minimum=1, maximum=3|" webui.py && \
 
48
  RUN git clone https://github.com/lllyasviel/Fooocus . && \
49
  # pin to a specific commit
50
  git checkout 18a8c6655208c31216b83bd93381cb12993597c2 && \
51
+ pip install --no-cache-dir xformers==0.0.20 torch==2.0.1 torchvision==0.15.2 --extra-index-url https://download.pytorch.org/whl/cu118 && \
52
+ pip install --no-cache-dir -r requirements_versions.txt
53
 
54
  # remove the next line if you're running on your own GPU, it set max images to 3 and disables the API
55
  RUN sed -i "s|image_number = gr.Slider(label='Image Number', minimum=1, maximum=32|image_number = gr.Slider(label='Image Number', minimum=1, maximum=3|" webui.py && \