tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

deprecate torch 2.7.1 (#3339)

Browse Source
This commit is contained in:
Wing Lian
2026-01-01 06:52:45 -05:00
committed by GitHub
parent 2b199f9915
commit afe18ace35
11 changed files with 65 additions and 144 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/axolotl/cli/cloud/baseten/template/train_sft.py
View File

@@ -24,8 +24,7 @@ if launcher_args:
launcher_args_str = "-- " + " ".join(launcher_args)
# 1. Define a base image for your training job
# must use torch 2.7.0 for vllm
BASE_IMAGE = "axolotlai/axolotl:main-py3.11-cu126-2.7.1"
BASE_IMAGE = "axolotlai/axolotl:main-py3.11-cu128-2.9.1"
# 2. Define the Runtime Environment for the Training Job
# This includes start commands and environment variables.a

2
src/axolotl/cli/cloud/modal_.py
View File

@@ -82,7 +82,7 @@ class ModalCloud(Cloud):
return res
def get_image(self):
docker_tag = "main-py3.11-cu126-2.7.1"
docker_tag = "main-py3.11-cu128-2.9.1"
if self.config.docker_tag:
docker_tag = self.config.docker_tag
docker_image = f"axolotlai/axolotl:{docker_tag}"