mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-05-10 04:06:49 +08:00
Move LogprobsOnePosition import to TYPE_CHECKING block in conftest.py
Signed-off-by: Chukwuma Nwaugha <nwaughac@gmail.com>
This commit is contained in:
parent
76802a010b
commit
101c957bff
@ -58,7 +58,6 @@ from vllm.distributed import (
|
||||
initialize_model_parallel,
|
||||
)
|
||||
from vllm.logger import init_logger
|
||||
from vllm.logprobs import LogprobsOnePosition
|
||||
from vllm.multimodal.utils import fetch_image
|
||||
from vllm.outputs import RequestOutput
|
||||
from vllm.sampling_params import BeamSearchParams
|
||||
@ -70,6 +69,7 @@ from torch._inductor.utils import fresh_cache
|
||||
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from vllm.logprobs import LogprobsOnePosition
|
||||
from transformers import PreTrainedTokenizer, PreTrainedTokenizerFast
|
||||
from transformers.generation.utils import GenerateOutput
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user