From a31746baa2143d15708a5da205d30525f1f47c81 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Sun, 7 May 2023 17:47:54 -0400 Subject: [PATCH] whoops, build from base image --- docker/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index b25aefccf..e896d0461 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -1,5 +1,5 @@ ARG BASE_TAG=main -FROM winglian/axolotl:$BASE_TAG +FROM winglian/axolotl-base:$BASE_TAG ARG TORCH_CUDA_ARCH_LIST="7.0 7.5 8.0 8.6+PTX"