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

Merge branch 'master' into release/1.20

Browse Source
This commit is contained in:
Jintao Huang
2024-11-08 15:51:08 +08:00
parent 47a8ba4d88 a393b6ef24
commit 1cb528d74f
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
docker/build_image.py
View File

@@ -23,17 +23,17 @@ class Builder:
# A mirrored image of nvidia/cuda:12.4.0-devel-ubuntu22.04
args.base_image = 'nvidia/cuda:12.1.0-devel-ubuntu22.04'
if not args.torch_version:
args.torch_version = '2.3.0'
args.torchaudio_version = '2.3.0'
args.torchvision_version = '0.18.0'
args.torch_version = '2.3.1'
args.torchaudio_version = '2.3.1'
args.torchvision_version = '0.18.1'
if not args.tf_version:
args.tf_version = '2.16.1'
if not args.cuda_version:
args.cuda_version = '12.1.0'
if not args.vllm_version:
args.vllm_version = '0.5.1'
args.vllm_version = '0.5.3'
if not args.lmdeploy_version:
args.lmdeploy_version = '0.5.0'
args.lmdeploy_version = '0.6.2'
if not args.autogptq_version:
args.autogptq_version = '0.7.1'
return args