Style

huggingface · Oct 26, 2023 · 30a922c · 30a922c
1 parent c26518b
commit 30a922c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/optimum/exporters/onnx/__main__.py b/optimum/exporters/onnx/__main__.py
@@ -340,7 +340,9 @@ def main_export(
             and kwargs_shapes.get(input_name) == 1
             and task.startswith("text-generation")
         ):
-            logger.warning("Exporting with a sequence length of 1 for text generation models is not supported and can yield unexpected results.")
+            logger.warning(
+                "Exporting with a sequence length of 1 for text generation models is not supported and can yield unexpected results."
+            )
 
     torch_dtype = None if fp16 is False else torch.float16