Skip to content

GitLab

Unverified Commit 31afe870 authored by bittersweet1999's avatar bittersweet1999 Committed by GitHub
Browse files

fix yi-chat template (#1178)

parent 1448be00
Hide 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