radames commited on
Commit
13331c6
1 Parent(s): 384acaf

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -46,8 +46,8 @@ WORKDIR $HOME/app
46
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
47
 
48
  RUN git clone https://github.com/comfyanonymous/ComfyUI . && \
49
- pip install --no-cache-dir -r requirements.txt
50
-
51
  # Checkpoints
52
 
53
  RUN echo "Downloading checkpoints..." && \
 
46
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
47
 
48
  RUN git clone https://github.com/comfyanonymous/ComfyUI . && \
49
+ pip install xformers!=0.0.18 --no-cache-dir -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu118 --extra-index-url https://download.pytorch.org/whl/cu117
50
+
51
  # Checkpoints
52
 
53
  RUN echo "Downloading checkpoints..." && \