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
f7c67f0e
Unverified
Commit
f7c67f0e
authored
Dec 16, 2023
by
Baber Abbasi
Committed by
GitHub
Dec 16, 2023
Browse files
`use_tqdm=False` if batch_size != auto (#1144)
parent
bd0f2414
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
2 deletions
+1
-2
lm_eval/models/vllm_causallms.py
lm_eval/models/vllm_causallms.py
+1
-2
No files found.
lm_eval/models/vllm_causallms.py
View file @
f7c67f0e
...
...
@@ -139,7 +139,6 @@ please install vllm via `pip install lm-eval[vllm]` or `pip install -e .[vllm]`"
generate
:
bool
=
False
,
max_tokens
:
int
=
None
,
stop
:
Optional
[
List
[
str
]]
=
None
,
use_tqdm
=
True
,
**
kwargs
,
):
if
"do_sample"
in
kwargs
.
keys
():
...
...
@@ -169,7 +168,7 @@ please install vllm via `pip install lm-eval[vllm]` or `pip install -e .[vllm]`"
outputs
=
self
.
model
.
generate
(
prompt_token_ids
=
requests
,
sampling_params
=
sampling_params
,
use_tqdm
=
use_tqdm
,
use_tqdm
=
True
if
self
.
batch_size
==
"auto"
else
False
,
)
return
outputs
...
...
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