Skip to content

Commit

Permalink
Update tests/openvino/test_modeling.py
Browse files Browse the repository at this point in the history
  • Loading branch information
echarlaix authored Oct 31, 2024
1 parent 43cca8f commit 3b0bf83
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/openvino/test_modeling.py
Original file line number Diff line number Diff line change
Expand Up @@ -933,7 +933,7 @@ def test_compare_to_transformers(self, model_arch):
transformers_outputs = transformers_model.generate(**tokens, generation_config=gen_config, **additional_inputs)
self.assertTrue(
torch.allclose(ov_outputs, transformers_outputs),
"OV output {ov_outputs}\nTransofrmers output {transformers_output}",
"OV output {ov_outputs}\nTransformers output {transformers_output}",
)

del transformers_model
Expand Down

0 comments on commit 3b0bf83

Please sign in to comment.