coderabbito

This commit is contained in:
Dan Saunders
2025-09-10 16:21:17 -04:00
parent cd8c769e84
commit 4c81172917
2 changed files with 9 additions and 4 deletions

View File

@@ -1,5 +1,7 @@
FROM axolotlai/axolotl-base-uv:{{ BASE_TAG }}
SHELL ["/bin/bash", "-euxo", "pipefail", "-c"]
ENV TORCH_CUDA_ARCH_LIST="7.0 7.5 8.0 8.6 9.0+PTX"
ENV AXOLOTL_EXTRAS="{{ AXOLOTL_EXTRAS }}"
ENV AXOLOTL_ARGS="{{ AXOLOTL_ARGS }}"