Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
lm-evaluation-harness
Commits
0c134ee9
Unverified
Commit
0c134ee9
authored
Sep 12, 2025
by
fxmarty-amd
Committed by
GitHub
Sep 12, 2025
Browse files
add quote to type hints (#3292)
parent
44398478
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
4 deletions
+4
-4
lm_eval/models/vllm_causallms.py
lm_eval/models/vllm_causallms.py
+4
-4
No files found.
lm_eval/models/vllm_causallms.py
View file @
0c134ee9
...
...
@@ -50,7 +50,7 @@ eval_logger = logging.getLogger(__name__)
def
_vllm_mp_worker
(
model_args
:
dict
,
sampling_params
:
"
list[SamplingParams
]
"
,
sampling_params
:
list
[
"
SamplingParams"
]
,
requests
:
list
[
list
[
int
]],
lora_request
:
"LoRARequest"
,
result_queue
:
"Queue"
,
...
...
@@ -364,7 +364,7 @@ class VLLM(TemplateLM):
self
,
requests
:
List
[
List
[
int
]]
=
None
,
generate
:
bool
=
False
,
sampling_params
:
Union
[
List
[
SamplingParams
],
SamplingParams
,
None
]
=
None
,
sampling_params
:
Union
[
List
[
"
SamplingParams
"
],
"
SamplingParams
"
,
None
]
=
None
,
):
if
not
generate
or
sampling_params
is
None
:
sampling_params
=
SamplingParams
(
...
...
@@ -379,9 +379,9 @@ class VLLM(TemplateLM):
@
ray
.
remote
def
run_inference_one_model
(
model_args
:
dict
,
sampling_params
:
List
[
SamplingParams
],
sampling_params
:
List
[
"
SamplingParams
"
],
requests
:
List
[
List
[
int
]],
lora_request
:
LoRARequest
,
lora_request
:
"
LoRARequest
"
,
):
llm
=
LLM
(
**
model_args
)
return
llm
.
generate
(
...
...
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