Skip to content

GitLab

Unverified Commit 8b2d3cbc authored by Simon Mo's avatar Simon Mo Committed by GitHub
Browse files

usage lib get version another way (#3735)

parent 9765b5c4
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
vllm/usage/usage_lib.py
View file @ 8b2d3cbc
...@@ -11,7 +11,6 @@ from typing import Dict, Optional ...@@ -11,7 +11,6 @@ from typing import Dict, Optional
from uuid import uuid4 from uuid import uuid4
import cpuinfo import cpuinfo
import pkg_resources
import psutil import psutil
import requests import requests
import torch import torch
...@@ -161,8 +160,9 @@ class UsageMessage: ...@@ -161,8 +160,9 @@ class UsageMessage:
]) ])
# vLLM information # vLLM information
import vllm # delayed import to prevent circular import
self.context = usage_context.value self.context = usage_context.value
self.vllm_version = pkg_resources.get_distribution("vllm").version self.vllm_version = vllm.__version__
self.model_architecture = model_architecture self.model_architecture = model_architecture
# Metadata # Metadata
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment