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
b177c82c
Unverified
Commit
b177c82c
authored
Feb 28, 2024
by
Linsong Chu
Committed by
GitHub
Feb 28, 2024
Browse files
fix duplicated kwargs in some model init (#1495)
parent
cc771eca
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
4 deletions
+4
-4
lm_eval/models/mamba_lm.py
lm_eval/models/mamba_lm.py
+3
-3
lm_eval/models/optimum_lm.py
lm_eval/models/optimum_lm.py
+1
-1
No files found.
lm_eval/models/mamba_lm.py
View file @
b177c82c
...
...
@@ -56,9 +56,9 @@ class MambaLMWrapper(HFLM):
super
().
__init__
(
pretrained
=
pretrained
,
# set appropriate defaults for tokenizer, max length, etc
backend
=
kwargs
.
get
(
"backend"
,
"causal"
),
tokenizer
=
kwargs
.
get
(
"tokenizer"
,
"EleutherAI/gpt-neox-20b"
),
max_length
=
kwargs
.
get
(
"max_length"
,
2048
),
backend
=
kwargs
.
pop
(
"backend"
,
"causal"
),
tokenizer
=
kwargs
.
pop
(
"tokenizer"
,
"EleutherAI/gpt-neox-20b"
),
max_length
=
kwargs
.
pop
(
"max_length"
,
2048
),
**
kwargs
,
)
...
...
lm_eval/models/optimum_lm.py
View file @
b177c82c
...
...
@@ -28,7 +28,7 @@ class OptimumLM(HFLM):
super
().
__init__
(
device
=
self
.
openvino_device
,
backend
=
kwargs
.
get
(
"backend"
,
"causal"
),
backend
=
kwargs
.
pop
(
"backend"
,
"causal"
),
**
kwargs
,
)
...
...
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