From 98a48c0967aecdbf0143d72822b3ba9b0f4335a6 Mon Sep 17 00:00:00 2001
From: Misha Chornyi <99709299+mc-nv@users.noreply.github.com>
Date: Thu, 26 Oct 2023 18:12:03 -0700
Subject: [PATCH] Update build.py

Co-authored-by: Olga Andreeva <124622579+oandreeva-nv@users.noreply.github.com>
---
 build.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/build.py b/build.py
index 805b187573..4fa1789fad 100755
--- 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=={}