From 48c777f4c3b61bbe73f11bf6e2bb887835eb92e4 Mon Sep 17 00:00:00 2001 From: hjh0119 Date: Mon, 7 Jul 2025 14:01:51 +0800 Subject: [PATCH] revert vllm&torch dep --- docker/build_image.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/docker/build_image.py b/docker/build_image.py index cade7adb..3f556ef3 100644 --- a/docker/build_image.py +++ b/docker/build_image.py @@ -269,13 +269,13 @@ 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.7.1' - args.torchaudio_version = '2.7.1' - args.torchvision_version = '0.22.1' + args.torch_version = '2.6.0' + args.torchaudio_version = '2.6.0' + args.torchvision_version = '0.21.0' if not args.cuda_version: args.cuda_version = '12.4.0' if not args.vllm_version: - args.vllm_version = '0.9.1' + args.vllm_version = '0.8.5.post1' if not args.lmdeploy_version: args.lmdeploy_version = '0.9.1' if not args.autogptq_version: