whoops, build from base image

2023-05-07 17:47:54 -04:00
parent 17345c8a4b
commit a31746baa2
1 changed files with 1 additions and 1 deletions
--- 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"