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
......@@ -11,7 +11,6 @@ from typing import Dict, Optional
from uuid import uuid4
import cpuinfo
import pkg_resources
import psutil
import requests
import torch
......@@ -161,8 +160,9 @@ class UsageMessage:
])
# vLLM information
import vllm # delayed import to prevent circular import
self.context = usage_context.value
self.vllm_version = pkg_resources.get_distribution("vllm").version
self.vllm_version = vllm.__version__
self.model_architecture = model_architecture
# 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