Skip to content

GitLab

"docker/vscode:/vscode.git/clone" did not exist on "264128cb9dbd83b666666945fd2fea0662135911"
Unverified Commit 31afe870 authored by bittersweet1999's avatar bittersweet1999 Committed by GitHub
Browse files

fix yi-chat template (#1178)

parent 1448be00
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
configs/models/yi/hf_yi_34b_chat.py
View file @ 31afe870
...@@ -8,5 +8,6 @@ models = [ ...@@ -8,5 +8,6 @@ models = [
max_out_len=1024, max_out_len=1024,
batch_size=8, batch_size=8,
run_cfg=dict(num_gpus=2), run_cfg=dict(num_gpus=2),
stop_words=['<|endoftext|>', '<|im_end|>'],
) )
] ]
configs/models/yi/hf_yi_6b_chat.py
View file @ 31afe870
...@@ -8,5 +8,6 @@ models = [ ...@@ -8,5 +8,6 @@ models = [
max_out_len=1024, max_out_len=1024,
batch_size=8, batch_size=8,
run_cfg=dict(num_gpus=1), run_cfg=dict(num_gpus=1),
stop_words=['<|endoftext|>', '<|im_end|>'],
) )
] ]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment