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
9142999e
Commit
9142999e
authored
Apr 15, 2023
by
Benjamin Fattori
Browse files
update type arguments for hf-causal to accept str for batch_size
parent
e87c0838
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
lm_eval/models/huggingface.py
lm_eval/models/huggingface.py
+1
-1
No files found.
lm_eval/models/huggingface.py
View file @
9142999e
...
@@ -69,7 +69,7 @@ class HuggingFaceAutoLM(BaseLM):
...
@@ -69,7 +69,7 @@ class HuggingFaceAutoLM(BaseLM):
tokenizer
:
Optional
[
str
]
=
None
,
tokenizer
:
Optional
[
str
]
=
None
,
subfolder
:
Optional
[
str
]
=
None
,
subfolder
:
Optional
[
str
]
=
None
,
revision
:
Optional
[
str
]
=
"main"
,
revision
:
Optional
[
str
]
=
"main"
,
batch_size
:
Optional
[
int
]
=
1
,
batch_size
:
Optional
[
Union
[
int
,
str
]
]
=
1
,
max_gen_toks
:
Optional
[
int
]
=
256
,
max_gen_toks
:
Optional
[
int
]
=
256
,
max_length
:
Optional
[
int
]
=
None
,
max_length
:
Optional
[
int
]
=
None
,
add_special_tokens
:
Optional
[
bool
]
=
None
,
add_special_tokens
:
Optional
[
bool
]
=
None
,
...
...
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