Fix xformers

ashleykleynhans · Mar 13, 2024 · a1002a6 · a1002a6
1 parent 69a18ee
commit a1002a6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -67,7 +67,7 @@ WORKDIR /
 RUN python3 -m venv /venv && \
  source /venv/bin/activate && \
  pip3 install torch==${TORCH_VERSION} --index-url ${INDEX_URL} && \
- pip3 install xformers==${XFORMERS_VERSION} --index-url ${INDEX_URL} && \
+ pip3 install xformers==${XFORMERS_VERSION} && \
  deactivate
 
 # Clone the git repo of Text Generation Web UI and set version

diff --git a/docker-bake.hcl b/docker-bake.hcl
@@ -21,7 +21,7 @@ target "default" {
  RELEASE = "${RELEASE}"
  INDEX_URL = "https://download.pytorch.org/whl/cu${CU_VERSION}"
  TORCH_VERSION = "2.2.0+cu${CU_VERSION}"
- XFORMERS_VERSION = "0.0.24+cu${CU_VERSION}"
+ XFORMERS_VERSION = "0.0.24"
  OOBABOOGA_COMMIT = "1934cb61ef879815644277c01c7295acbae542d8"
  RUNPODCTL_VERSION = "v1.14.2"
  VENV_PATH = "/workspace/venvs/text-generation-webuii"