Unverified Commit 384dc7f7 authored by Flora Feng's avatar Flora Feng Committed by GitHub
Browse files

[Refactor] Relocate completion and chat completion tests (#37125)


Signed-off-by: default avatarsfeng33 <4florafeng@gmail.com>
parent f04d5226
...@@ -14,7 +14,7 @@ import torch ...@@ -14,7 +14,7 @@ import torch
from openai import BadRequestError from openai import BadRequestError
from transformers import AutoConfig from transformers import AutoConfig
from ...utils import RemoteOpenAIServer from tests.utils import RemoteOpenAIServer
# any model with a chat template should work here # any model with a chat template should work here
MODEL_NAME = "facebook/opt-125m" MODEL_NAME = "facebook/opt-125m"
......
...@@ -11,11 +11,10 @@ import pytest ...@@ -11,11 +11,10 @@ import pytest
import regex as re import regex as re
import torch import torch
from tests.utils import RemoteOpenAIServer
from vllm.config import ModelConfig from vllm.config import ModelConfig
from vllm.renderers.embed_utils import safe_load_prompt_embeds from vllm.renderers.embed_utils import safe_load_prompt_embeds
from ...utils import RemoteOpenAIServer
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_empty_prompt(): async def test_empty_prompt():
......
...@@ -9,6 +9,7 @@ import pytest ...@@ -9,6 +9,7 @@ import pytest
import pytest_asyncio import pytest_asyncio
import torch.cuda import torch.cuda
from tests.utils import RemoteOpenAIServer
from vllm.engine.arg_utils import EngineArgs from vllm.engine.arg_utils import EngineArgs
from vllm.model_executor.model_loader.tensorizer import ( from vllm.model_executor.model_loader.tensorizer import (
TensorizerConfig, TensorizerConfig,
...@@ -17,8 +18,6 @@ from vllm.model_executor.model_loader.tensorizer import ( ...@@ -17,8 +18,6 @@ from vllm.model_executor.model_loader.tensorizer import (
) )
from vllm.platforms import current_platform from vllm.platforms import current_platform
from ...utils import RemoteOpenAIServer
MODEL_NAME = "unsloth/llama-3.2-1b-Instruct" MODEL_NAME = "unsloth/llama-3.2-1b-Instruct"
LORA_PATH = "davzoku/finqa_adapter_1b" LORA_PATH = "davzoku/finqa_adapter_1b"
......
...@@ -6,11 +6,10 @@ import tempfile ...@@ -6,11 +6,10 @@ import tempfile
import pytest import pytest
from tests.utils import RemoteOpenAIServer
from vllm.model_executor.model_loader.weight_utils import download_weights_from_hf from vllm.model_executor.model_loader.weight_utils import download_weights_from_hf
from vllm.tokenizers import get_tokenizer from vllm.tokenizers import get_tokenizer
from ...utils import RemoteOpenAIServer
MODEL_NAME = "Qwen/Qwen3-0.6B" MODEL_NAME = "Qwen/Qwen3-0.6B"
MODEL_PATH = os.path.join(tempfile.gettempdir(), "qwen3_06b") MODEL_PATH = os.path.join(tempfile.gettempdir(), "qwen3_06b")
......
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