From a1002a60beeafc70b6c3f5492d2c6d7c59750d91 Mon Sep 17 00:00:00 2001
From: Ashley Kleynhans <ashley.kleynhans@gmail.com>
Date: Wed, 13 Mar 2024 14:29:15 +0200
Subject: [PATCH] Fix xformers

---
 Dockerfile      | 2 +-
 docker-bake.hcl | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 9e03058..bce34ce 100644
--- 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
index 9da6bc8..7dac651 100644
--- 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"