Skip to content

Commit

Permalink
Bumped to latest oobabooga and bumped runpodctl to version 1.14.2
Browse files Browse the repository at this point in the history
  • Loading branch information
ashleykleynhans committed Mar 8, 2024
1 parent fd0f822 commit c1fe138
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions Dockerfile
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=60f3d87309bd5fa8e3d77ed1fc66b25ef84db8c5
ARG OOBABOOGA_COMMIT=aa0da07af012e251b4b70a1391b0acd360f796bd
ARG TORCH_VERSION=2.1.2
ARG XFORMERS_VERSION=0.0.23.post1

Expand Down Expand Up @@ -96,7 +96,8 @@ RUN source /venv/bin/activate && \
RUN curl https://rclone.org/install.sh | bash

# Install runpodctl
RUN wget https://github.com/runpod/runpodctl/releases/download/v1.13.0/runpodctl-linux-amd64 -O runpodctl && \
ARG RUNPODCTL_VERSION="v1.14.2"
RUN wget "https://github.com/runpod/runpodctl/releases/download/${RUNPODCTL_VERSION}/runpodctl-linux-amd64" -O runpodctl && \
chmod a+x runpodctl && \
mv runpodctl /usr/local/bin

Expand Down

0 comments on commit c1fe138

Please sign in to comment.