Skip to content

Commit a80b073

Browse files
committed
fix code style
1 parent 0551064 commit a80b073

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/ipex/test_modeling.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -269,7 +269,9 @@ def test_assisted_decoding(self, model_arch):
269269
ipex_output = ipex_model.generate(**tokens, do_sample=False)
270270
ipex_output_assisted = ipex_model.generate(**tokens, do_sample=False, assistant_model=transformers_model)
271271
transformers_output = transformers_model.generate(**tokens, do_sample=False)
272-
transformers_output_assisted = transformers_model.generate(**tokens, do_sample=False, assistant_model=ipex_model)
272+
transformers_output_assisted = transformers_model.generate(
273+
**tokens, do_sample=False, assistant_model=ipex_model
274+
)
273275
self.assertTrue(torch.equal(ipex_output, ipex_output_assisted))
274276
self.assertTrue(torch.equal(transformers_output, transformers_output_assisted))
275277

0 commit comments

Comments
 (0)