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
OpenDAS
opencompass
Commits
50b658d2
Unverified
Commit
50b658d2
authored
Jul 11, 2023
by
Leymore
Committed by
GitHub
Jul 11, 2023
Browse files
[Fix] Update HF configs (#42)
parent
0625294e
Changes
7
Hide whitespace changes
Inline
Side-by-side
Showing
7 changed files
with
37 additions
and
4 deletions
+37
-4
configs/models/hf_internlm_7b.py
configs/models/hf_internlm_7b.py
+3
-2
configs/models/hf_internlm_chat_7b.py
configs/models/hf_internlm_chat_7b.py
+3
-2
configs/models/hf_internlm_chat_7b_8k.py
configs/models/hf_internlm_chat_7b_8k.py
+30
-0
configs/models/hf_llama_13b.py
configs/models/hf_llama_13b.py
+1
-0
configs/models/hf_vicuna_13b.py
configs/models/hf_vicuna_13b.py
+0
-0
configs/models/hf_vicuna_33b.py
configs/models/hf_vicuna_33b.py
+0
-0
configs/models/hf_vicuna_7b.py
configs/models/hf_vicuna_7b.py
+0
-0
No files found.
configs/models/hf_internlm_7b.py
View file @
50b658d2
...
@@ -5,12 +5,13 @@ models = [
...
@@ -5,12 +5,13 @@ models = [
dict
(
dict
(
type
=
HuggingFaceCausalLM
,
type
=
HuggingFaceCausalLM
,
abbr
=
'internlm-7b-hf'
,
abbr
=
'internlm-7b-hf'
,
path
=
"internlm-7b"
,
path
=
"internlm
/internlm
-7b"
,
tokenizer_path
=
'internlm-7b'
,
tokenizer_path
=
'internlm
/internlm
-7b'
,
tokenizer_kwargs
=
dict
(
tokenizer_kwargs
=
dict
(
padding_side
=
'left'
,
padding_side
=
'left'
,
truncation_side
=
'left'
,
truncation_side
=
'left'
,
use_fast
=
False
,
use_fast
=
False
,
trust_remote_code
=
True
,
),
),
max_out_len
=
100
,
max_out_len
=
100
,
max_seq_len
=
2048
,
max_seq_len
=
2048
,
...
...
configs/models/hf_internlm_chat_7b.py
View file @
50b658d2
...
@@ -12,12 +12,13 @@ models = [
...
@@ -12,12 +12,13 @@ models = [
dict
(
dict
(
type
=
HuggingFaceCausalLM
,
type
=
HuggingFaceCausalLM
,
abbr
=
'internlm-chat-7b-hf'
,
abbr
=
'internlm-chat-7b-hf'
,
path
=
"internlm-chat-7b"
,
path
=
"internlm
/internlm
-chat-7b"
,
tokenizer_path
=
'internlm-chat-7b'
,
tokenizer_path
=
'internlm
/internlm
-chat-7b'
,
tokenizer_kwargs
=
dict
(
tokenizer_kwargs
=
dict
(
padding_side
=
'left'
,
padding_side
=
'left'
,
truncation_side
=
'left'
,
truncation_side
=
'left'
,
use_fast
=
False
,
use_fast
=
False
,
trust_remote_code
=
True
,
),
),
max_out_len
=
100
,
max_out_len
=
100
,
max_seq_len
=
2048
,
max_seq_len
=
2048
,
...
...
configs/models/hf_internlm_chat_7b_8k.py
0 → 100644
View file @
50b658d2
from
opencompass.models
import
HuggingFaceCausalLM
_meta_template
=
dict
(
round
=
[
dict
(
role
=
'HUMAN'
,
begin
=
'<|User|>:'
,
end
=
'<eoh>
\n
'
),
dict
(
role
=
'BOT'
,
begin
=
'<|Bot|>:'
,
end
=
'<eoa>
\n
'
,
generate
=
True
),
],
)
models
=
[
dict
(
type
=
HuggingFaceCausalLM
,
abbr
=
'internlm-chat-7b-8k-hf'
,
path
=
"internlm/internlm-chat-7b-8k"
,
tokenizer_path
=
'internlm/internlm-chat-7b-8k'
,
tokenizer_kwargs
=
dict
(
padding_side
=
'left'
,
truncation_side
=
'left'
,
use_fast
=
False
,
trust_remote_code
=
True
,
),
max_out_len
=
100
,
max_seq_len
=
2048
,
batch_size
=
8
,
meta_template
=
_meta_template
,
model_kwargs
=
dict
(
trust_remote_code
=
True
,
device_map
=
'auto'
),
run_cfg
=
dict
(
num_gpus
=
1
,
num_procs
=
1
),
)
]
configs/models/hf_llama_13b.py
View file @
50b658d2
...
@@ -5,6 +5,7 @@ models = [
...
@@ -5,6 +5,7 @@ models = [
# LLaMA 13B
# LLaMA 13B
dict
(
dict
(
type
=
HuggingFaceCausalLM
,
type
=
HuggingFaceCausalLM
,
abbr
=
'llama-13b-hf'
,
path
=
"decapoda-research/llama-13b-hf"
,
path
=
"decapoda-research/llama-13b-hf"
,
tokenizer_path
=
'decapoda-research/llama-13b-hf'
,
tokenizer_path
=
'decapoda-research/llama-13b-hf'
,
tokenizer_kwargs
=
dict
(
padding_side
=
'left'
,
tokenizer_kwargs
=
dict
(
padding_side
=
'left'
,
...
...
configs/models/hf_vicuna_
v1.3_
13b.py
→
configs/models/hf_vicuna_13b.py
View file @
50b658d2
File moved
configs/models/hf_vicuna_
v1.3_
33b.py
→
configs/models/hf_vicuna_33b.py
View file @
50b658d2
File moved
configs/models/hf_vicuna_
v1.3_
7b.py
→
configs/models/hf_vicuna_7b.py
View file @
50b658d2
File moved
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