From 83c70cd3562ba54b5dd65db6e5c604e941316b33 Mon Sep 17 00:00:00 2001 From: Nilabhra Date: Mon, 13 May 2024 10:26:51 +0000 Subject: [PATCH] chore: formating. --- tests/models/falcon/test_modeling_falcon.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/models/falcon/test_modeling_falcon.py b/tests/models/falcon/test_modeling_falcon.py index ab955dae6ad301..5bc22f31e8596f 100644 --- a/tests/models/falcon/test_modeling_falcon.py +++ b/tests/models/falcon/test_modeling_falcon.py @@ -610,9 +610,9 @@ def test_lm_generate_falcon_11b(self): "tiiuae/falcon-11B", device_map={"": torch_device}, load_in_8bit=True ) model.eval() - inputs = tokenizer("Two roads diverged in a yellow wood,", return_tensors="pt", return_token_type_ids=False).to( - torch_device - ) + inputs = tokenizer( + "Two roads diverged in a yellow wood,", return_tensors="pt", return_token_type_ids=False + ).to(torch_device) EXPECTED_OUTPUT = "Two roads diverged in a yellow wood,\nAnd sorry I could not travel both\n"