✨ add console scripts to setup.py

Signed-off-by: Prashant Gupta <prashantgupta@us.ibm.com>

pyproject.toml

@@ -8,13 +8,9 @@ requires = [
     "torch == 2.2.1",
     "wheel",
     "typer == 0.9.*",
-    "loguru == 0.7.*"
 ]
 build-backend = "setuptools.build_meta"
 
-[project.scripts]
-vllm = 'vllm.cli:app'
-
 [tool.ruff]
 # Allow lines to be as long as 80.
 line-length = 80

requirements-build.txt

@@ -5,5 +5,4 @@ packaging
 setuptools>=49.4.0
 torch==2.2.1
 wheel
-typer==0.9.*
-loguru==0.7.*
+typer==0.9.*

setup.py

@@ -406,4 +406,7 @@ def _read_requirements(filename: str) -> List[str]:
     ext_modules=ext_modules,
     cmdclass={"build_ext": cmake_build_ext} if not _is_neuron() else {},
     package_data=package_data,
+    entry_points={
+        "console_scripts": ["vllm=vllm.cli:app"],
+    },
 )

vllm/tgis_utils/hub.py

@@ -2,6 +2,7 @@
 import datetime
 import glob
 import json
+import logging
 import os
 from collections import defaultdict
 from concurrent.futures import ThreadPoolExecutor
@@ -12,12 +13,12 @@
 import torch
 from huggingface_hub import HfApi, hf_hub_download, try_to_load_from_cache
 from huggingface_hub.utils import LocalEntryNotFoundError
-from loguru import logger
 from safetensors.torch import (_find_shared_tensors, _is_complete, load_file,
                                save_file)
 from tqdm import tqdm
 
 TRUST_REMOTE_CODE = os.getenv("TRUST_REMOTE_CODE") == "true"
+logger = logging.getLogger(__name__)
 
 
 def weight_hub_files(model_name,