Skip to content

Commit 68c68d7

Browse files
authored
Set missing baseline values for text_generation tests. (#2316)
Signed-off-by: Artur Kloniecki <arturx.kloniecki@intel.com>
1 parent 97d958a commit 68c68d7

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tests/baselines/fixture/tests/test_text_generation_example.json

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -829,12 +829,20 @@
829829
"gaudi2": {
830830
"output": "DeepSpeed is a machine learning framework that makes distributed training easy, efficient, and flexible. DeepSpeed can train BERT-Large on",
831831
"throughput": 0.7572952
832+
},
833+
"gaudi3": {
834+
"output": "DeepSpeed is a machine learning framework that makes distributed training easy, efficient, and effective. It is a deep learning optimization library that makes",
835+
"throughput": 0.2239626836276946
832836
}
833837
},
834838
"tests/test_text_generation_example.py::test_text_generation_bnb[meta-llama/Llama-3.1-70B-1-20-True-True]": {
835839
"gaudi2": {
836840
"output": "DeepSpeed is a machine learning framework that makes distributed training easy, efficient, and effective. It is a deep learning optimization library that makes",
837841
"throughput": 0.7583387
842+
},
843+
"gaudi3": {
844+
"output": "DeepSpeed is a machine learning framework that makes distributed training easy, efficient, and flexible. DeepSpeed can train BERT-Large on",
845+
"throughput": 0.22355754926812374
838846
}
839847
}
840848
}

0 commit comments

Comments
 (0)