Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
f408d05c
"vscode:/vscode.git/clone" did not exist on "3693f922ff0e15b1e395cdc862f7c6e492a11bf4"
Unverified
Commit
f408d05c
authored
Mar 25, 2024
by
Simon Mo
Committed by
GitHub
Mar 25, 2024
Browse files
hotfix isort on logprobs ranks pr (#3622)
parent
0b4997e0
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
1 deletion
+2
-1
tests/samplers/test_ranks.py
tests/samplers/test_ranks.py
+1
-0
tests/samplers/test_sampler.py
tests/samplers/test_sampler.py
+1
-1
No files found.
tests/samplers/test_ranks.py
View file @
f408d05c
import
pytest
from
vllm
import
SamplingParams
MODELS
=
[
"facebook/opt-125m"
]
...
...
tests/samplers/test_sampler.py
View file @
f408d05c
...
...
@@ -9,8 +9,8 @@ from transformers import GenerationConfig, GenerationMixin
from
vllm.model_executor.layers.sampler
import
Sampler
from
vllm.model_executor.utils
import
set_random_seed
from
vllm.sequence
import
SamplingParams
,
SequenceData
,
SequenceGroupMetadata
from
vllm.worker.model_runner
import
ModelRunner
from
vllm.utils
import
Counter
from
vllm.worker.model_runner
import
ModelRunner
class
MockLogitsSampler
(
Sampler
):
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment