usage lib get version another way (#3735)
This commit is contained in:
parent
9765b5c406
commit
8b2d3cbc1b
@ -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
|
||||
|
||||
Loading…
Reference in New Issue
Block a user