Skip to content

Commit fc2405c

Browse files
authored
[gaudi] Fix the CI test errors (#3286)
Signed-off-by: yuanwu <yuan.wu@intel.com>
1 parent ebb26f0 commit fc2405c

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

integration-tests/gaudi/test_gaudi_generate.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -102,8 +102,8 @@ def pytest_configure(config):
102102
"google/gemma-7b-it": {
103103
"model_id": "google/gemma-7b-it",
104104
"input": "What is Deep Learning?",
105-
"expected_greedy_output": "\n\nDeep learning is a subset of machine learning that uses artificial neural networks to learn from large amounts of data. Neural networks are inspired by the structure and function of",
106-
"expected_batch_output": "\n\nDeep learning is a subset of machine learning that uses artificial neural networks to learn from large amounts of data. Neural networks are inspired by the structure and function of",
105+
"expected_greedy_output": "\n\nDeep learning is a subset of machine learning that uses artificial neural networks to learn from large amounts of data. Deep learning is a powerful tool for many tasks,",
106+
"expected_batch_output": "\n\nDeep learning is a subset of machine learning that uses artificial neural networks to learn from large amounts of data. Deep learning is a powerful tool for many tasks,",
107107
"args": [
108108
"--max-input-tokens",
109109
"512",
@@ -162,8 +162,8 @@ def pytest_configure(config):
162162
"openai-community/gpt2": {
163163
"model_id": "openai-community/gpt2",
164164
"input": "What is Deep Learning?",
165-
"expected_greedy_output": "\n\nDeep learning is a new field of research that has been around for a long time. It is a new field of research that has been around for a",
166-
"expected_batch_output": "\n\nDeep learning is a new field of research that has been around for a long time. It is a new field of research that has been around for a",
165+
"expected_greedy_output": "\n\nDeep learning is a subset of machine learning that is based on artificial neural networks. It is a type of machine learning that is based on the idea of",
166+
"expected_batch_output": "\n\nDeep learning is a subset of machine learning that is based on artificial neural networks. It is a type of machine learning that is based on the idea of",
167167
"args": [
168168
"--max-input-tokens",
169169
"512",

0 commit comments

Comments
 (0)