From 9ab2825015d9c6a13a5959d2b17d8fe9cb9ccaef Mon Sep 17 00:00:00 2001 From: "yuze.zyz" Date: Mon, 10 Feb 2025 10:52:02 +0800 Subject: [PATCH] Revert "update version" This reverts commit d501fd405b82ef7070250f0819fe12338e4cf903. --- docker/build_image.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/docker/build_image.py b/docker/build_image.py index c46a2f88..344fc9d3 100644 --- a/docker/build_image.py +++ b/docker/build_image.py @@ -263,15 +263,15 @@ 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.5.1' - args.torchaudio_version = '2.5.1' - args.torchvision_version = '0.20.1' + args.torch_version = '2.4.0' + args.torchaudio_version = '2.4.0' + args.torchvision_version = '0.19.0' if not args.cuda_version: args.cuda_version = '12.4.0' if not args.vllm_version: - args.vllm_version = '0.7.2' + args.vllm_version = '0.6.3.post1' if not args.lmdeploy_version: - args.lmdeploy_version = '0.7.0.post2' + args.lmdeploy_version = '0.6.2' if not args.autogptq_version: args.autogptq_version = '0.7.1' return args