We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0551064 commit a80b073Copy full SHA for a80b073
tests/ipex/test_modeling.py
@@ -269,7 +269,9 @@ def test_assisted_decoding(self, model_arch):
269
ipex_output = ipex_model.generate(**tokens, do_sample=False)
270
ipex_output_assisted = ipex_model.generate(**tokens, do_sample=False, assistant_model=transformers_model)
271
transformers_output = transformers_model.generate(**tokens, do_sample=False)
272
- transformers_output_assisted = transformers_model.generate(**tokens, do_sample=False, assistant_model=ipex_model)
+ transformers_output_assisted = transformers_model.generate(
273
+ **tokens, do_sample=False, assistant_model=ipex_model
274
+ )
275
self.assertTrue(torch.equal(ipex_output, ipex_output_assisted))
276
self.assertTrue(torch.equal(transformers_output, transformers_output_assisted))
277
0 commit comments