tharms commited on
Commit
864416c
·
1 Parent(s): b66adbc

CUDA incompatible

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -10,7 +10,7 @@ RUN apt-get update -y && \
10
  COPY requirements_docker.txt requirements_versions.txt /tmp/
11
  RUN pip install --no-cache-dir -r /tmp/requirements_docker.txt -r /tmp/requirements_versions.txt && \
12
  rm -f /tmp/requirements_docker.txt /tmp/requirements_versions.txt
13
- RUN pip install --no-cache-dir xformers==0.0.23 --no-dependencies
14
  RUN curl -fsL -o /usr/local/lib/python3.10/dist-packages/gradio/frpc_linux_amd64_v0.2 https://cdn-media.huggingface.co/frpc-gradio-0.2/frpc_linux_amd64 && \
15
  chmod +x /usr/local/lib/python3.10/dist-packages/gradio/frpc_linux_amd64_v0.2
16
 
 
10
  COPY requirements_docker.txt requirements_versions.txt /tmp/
11
  RUN pip install --no-cache-dir -r /tmp/requirements_docker.txt -r /tmp/requirements_versions.txt && \
12
  rm -f /tmp/requirements_docker.txt /tmp/requirements_versions.txt
13
+ RUN pip install -U xformers --index-url https://download.pytorch.org/whl/cu121
14
  RUN curl -fsL -o /usr/local/lib/python3.10/dist-packages/gradio/frpc_linux_amd64_v0.2 https://cdn-media.huggingface.co/frpc-gradio-0.2/frpc_linux_amd64 && \
15
  chmod +x /usr/local/lib/python3.10/dist-packages/gradio/frpc_linux_amd64_v0.2
16