Update build.py

Co-authored-by: Olga Andreeva <124622579+oandreeva-nv@users.noreply.github.com>
triton-inference-server · Oct 27, 2023 · 98a48c0 · 98a48c0
1 parent 6200f5b
commit 98a48c0
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/build.py b/build.py
@@ -1357,8 +1357,6 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach
     if "vllm" in backends:
         # [DLIS-5606] Build Conda environment for vLLM backend
         # Remove Pip install once vLLM backend moves to Conda environment.
-        # [DLIS-5650] Pre-installing torch 2.0.1, since vllm 0.2
-        # requires torch >= 2.0.0, but it doesn't work with torch 2.1.0.
         df += """
 # vLLM needed for vLLM backend
 RUN pip3 install torch==2.0.1 vllm=={}