Skip to content

Commit

Permalink
Bump to latest version of Oobabooga
Browse files Browse the repository at this point in the history
  • Loading branch information
@ashleykleynhans
ashleykleynhans committed Mar 11, 2024
1 parent 6e5aa05 commit bd96da1
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=aa0da07af012e251b4b70a1391b0acd360f796bd
ARG OOBABOOGA_COMMIT=1934cb61ef879815644277c01c7295acbae542d8
ARG TORCH_VERSION=2.2.0
ARG XFORMERS_VERSION=0.0.24

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

# Set template version
ENV TEMPLATE_VERSION=1.13.0
ENV TEMPLATE_VERSION=1.13.1

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

0 comments on commit bd96da1

Please sign in to comment.