Skip to content

Commit 012cb96

Browse files
Yejing-Laiblzheng
andauthored
add baichuan2 to deepspeed accuracy script (#3454)
Co-authored-by: blzheng <[email protected]>
1 parent 568fbc1 commit 012cb96

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

examples/cpu/llm/inference/distributed/run_accuracy_with_deepspeed.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -327,6 +327,7 @@ def __init__(
327327
if world_size == 1 or model_type in [
328328
"falcon",
329329
"baichuan",
330+
"baichuan2",
330331
"gptbigcode",
331332
"qwen",
332333
"yuan",

0 commit comments

Comments
 (0)