Skip to content

Commit

Permalink
Remove xformers
Browse files Browse the repository at this point in the history
  • Loading branch information
ashleykleynhans committed Mar 13, 2024
1 parent b3e4b53 commit 9dc798b
Show file tree
Hide file tree
Showing 3 changed files with 0 additions and 4 deletions.
2 changes: 0 additions & 2 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -62,12 +62,10 @@ FROM base as setup
# Install Torch
ARG INDEX_URL
ARG TORCH_VERSION
ARG XFORMERS_VERSION
WORKDIR /
RUN python3 -m venv /venv && \
source /venv/bin/activate && \
pip3 install torch==${TORCH_VERSION} --index-url ${INDEX_URL} && \
pip3 install xformers==${XFORMERS_VERSION} && \
deactivate

# Clone the git repo of Text Generation Web UI and set version
Expand Down
1 change: 0 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,6 @@
* [Text Generation Web UI](
https://github.com/oobabooga/text-generation-webui)
* Torch 2.2.1
* xformers 0.0.24
* Jupyter Lab
* [runpodctl](https://github.com/runpod/runpodctl)
* [OhMyRunPod](https://github.com/kodxana/OhMyRunPod)
Expand Down
1 change: 0 additions & 1 deletion docker-bake.hcl
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,6 @@ target "default" {
RELEASE = "${RELEASE}"
INDEX_URL = "https://download.pytorch.org/whl/cu${CU_VERSION}"
TORCH_VERSION = "2.2.1+cu${CU_VERSION}"
XFORMERS_VERSION = "0.0.24"
OOBABOOGA_COMMIT = "1934cb61ef879815644277c01c7295acbae542d8"
RUNPODCTL_VERSION = "v1.14.2"
VENV_PATH = "/workspace/venvs/text-generation-webuii"
Expand Down

0 comments on commit 9dc798b

Please sign in to comment.