phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix vllm import (#7224)

Browse source 
* fix: Fix vllm import

* Update handler.py
This commit is contained in:
Ivan Vykopal 2024-12-15 00:57:49 +01:00 • committed by GitHub
parent 0038dc1762
commit a091964dd4
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/llms/vllm/completion/handler.py
View file

@ -29,8 +29,7 @@ class VLLMError(Exception):
def validate_environment(model: str): def validate_environment(model: str):
global llm global llm
try: try:
from litellm.llms.vllm.completion.handler import LLM # type: ignore from vllm import LLM, SamplingParams # type: ignore
from litellm.llms.vllm.completion.handler import SamplingParams # type: ignore
if llm is None: if llm is None:
llm = LLM(model=model) llm = LLM(model=model)