Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-09-01 19:49:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

update docker image for 1.28

Browse Source
This commit is contained in:
hjh0119
2025-07-07 10:34:30 +08:00
parent ac7404ce87
commit 5a8f97cbd1
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
docker/build_image.py
View File

@@ -269,19 +269,19 @@ class LLMImageBuilder(Builder):
# A mirrored image of nvidia/cuda:12.4.0-devel-ubuntu22.04
args.base_image = 'nvidia/cuda:12.4.0-devel-ubuntu22.04'
if not args.torch_version:
args.torch_version = '2.6.0'
args.torchaudio_version = '2.6.0'
args.torchvision_version = '0.21.0'
args.torch_version = '2.7.1'
args.torchaudio_version = '2.7.1'
args.torchvision_version = '0.22.1'
if not args.cuda_version:
args.cuda_version = '12.4.0'
if not args.vllm_version:
args.vllm_version = '0.8.5.post1'
args.vllm_version = '0.9.1'
if not args.lmdeploy_version:
args.lmdeploy_version = '0.7.2.post1'
args.lmdeploy_version = '0.9.1'
if not args.autogptq_version:
args.autogptq_version = '0.7.1'
if not args.flashattn_version:
args.flashattn_version = '2.7.1.post4'
args.flashattn_version = '2.8.0.post2'
return args
def generate_dockerfile(self) -> str: