diff --git a/docker/build_image.py b/docker/build_image.py index 344fc9d3..8ba4fd68 100644 --- a/docker/build_image.py +++ b/docker/build_image.py @@ -284,7 +284,7 @@ class LLMImageBuilder(Builder): self.args.python_version) version_args = ( f'{self.args.torch_version} {self.args.torchvision_version} {self.args.torchaudio_version} ' - f'{self.args.vllm_version} {self.args.lmdeploy_version} {self.args.autogptq_version}' + f'{self.args.lmdeploy_version} {self.args.vllm_version} {self.args.autogptq_version}' ) with open('docker/Dockerfile.ubuntu', 'r') as f: content = f.read()