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

feat(modal): update docker tag to use torch2.6 from torch2.5 (#2749) [skip ci]

Browse Source
This commit is contained in:
NanoCode012
2025-06-03 14:26:07 -07:00
committed by GitHub
parent d7fa60662e
commit 4b1a29c694
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

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