make fixup

huggingface · Sep 14, 2023 · 0be6f03 · 0be6f03
1 parent 52d0904
commit 0be6f03
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/transformers/pipelines/conversational.py b/src/transformers/pipelines/conversational.py
@@ -275,7 +275,9 @@ def _forward(self, model_inputs, minimum_tokens=10, **generate_kwargs):
 
         n = model_inputs["input_ids"].shape[1]
         if max_length - minimum_tokens < n:
-            logger.warning(f"Conversation input is too long ({n}), trimming it to {max_length - minimum_tokens} tokens. Consider increasing `max_length` to avoid truncation.")
+            logger.warning(
+                f"Conversation input is too long ({n}), trimming it to {max_length - minimum_tokens} tokens. Consider increasing `max_length` to avoid truncation."
+            )
             trim = max_length - minimum_tokens
             model_inputs["input_ids"] = model_inputs["input_ids"][:, -trim:]
             if "attention_mask" in model_inputs: