chore: formating.

huggingface · May 13, 2024 · 83c70cd · 83c70cd
1 parent 92e0993
commit 83c70cd
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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"