Skip to content

Commit

Permalink
Bump to latest Oobabooga version
Browse files Browse the repository at this point in the history
  • Loading branch information
@ashleykleynhans
ashleykleynhans committed Mar 4, 2024
1 parent 311a61a commit fd0f822
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions Dockerfile
View file
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
# Stage 1: Base
FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04 as base

ARG OOBABOOGA_COMMIT=ba852716fdc9be64710544b270e017eb1e766373
ARG OOBABOOGA_COMMIT=60f3d87309bd5fa8e3d77ed1fc66b25ef84db8c5
ARG TORCH_VERSION=2.1.2
ARG XFORMERS_VERSION=0.0.23.post1

Expand Down Expand Up @@ -135,7 +135,7 @@ COPY fetch_model.py /text-generation-webui/
COPY download_model.py /text-generation-webui/

# Set template version
ENV TEMPLATE_VERSION=1.12.6
ENV TEMPLATE_VERSION=1.12.7

# Set the venv path
ENV VENV_PATH="/workspace/venvs/text-generation-webui"
Expand Down

0 comments on commit fd0f822

Please sign in to comment.