xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-13 22:45:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

hotfix isort on logprobs ranks pr (#3622)

Browse Source
This commit is contained in:
Simon Mo 2024-03-25 11:55:46 -07:00 committed by GitHub
parent 0b4997e05c
commit f408d05c52
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/samplers/test_ranks.py
View File

@ -1,4 +1,5 @@
import pytest import pytest
from vllm import SamplingParams from vllm import SamplingParams
MODELS = ["facebook/opt-125m"] MODELS = ["facebook/opt-125m"]

2
tests/samplers/test_sampler.py
View File

@ -9,8 +9,8 @@ from transformers import GenerationConfig, GenerationMixin
from vllm.model_executor.layers.sampler import Sampler from vllm.model_executor.layers.sampler import Sampler
from vllm.model_executor.utils import set_random_seed from vllm.model_executor.utils import set_random_seed
from vllm.sequence import SamplingParams, SequenceData, SequenceGroupMetadata from vllm.sequence import SamplingParams, SequenceData, SequenceGroupMetadata
from vllm.worker.model_runner import ModelRunner
from vllm.utils import Counter from vllm.utils import Counter
from vllm.worker.model_runner import ModelRunner
class MockLogitsSampler(Sampler): class MockLogitsSampler(Sampler):