Apply suggestions from code review

Signed-off-by: Ettore Di Giacinto <[email protected]>
mudler · Oct 4, 2024 · d68e7ae · d68e7ae
1 parent 3d737d1
commit d68e7ae
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/backend/python/vllm/backend.py b/backend/python/vllm/backend.py
@@ -284,7 +284,7 @@ def load_video(self, video_path: str):
             video_path (str): The path to the image file.
 
         Returns:
-            Image: The loaded image.
+            Video: The loaded video.
         """
         try:
             video = VideoAsset(name=video_path).np_ndarrays

diff --git a/backend/python/vllm/install.sh b/backend/python/vllm/install.sh
@@ -16,7 +16,6 @@ fi
 if [ "x${BUILD_TYPE}" == "x" ]; then
         ensureVenv
         # https://docs.vllm.ai/en/v0.6.1/getting_started/cpu-installation.html
-        export VLLM_TARGET_DEVICE=cpu
         if [ ! -d vllm ]; then
             git clone https://github.com/vllm-project/vllm
         fi