From 30a922c450a452d9a1f375451c714a20aaa17aa7 Mon Sep 17 00:00:00 2001 From: baskrahmer Date: Thu, 26 Oct 2023 16:32:38 +0000 Subject: [PATCH] Style --- optimum/exporters/onnx/__main__.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/optimum/exporters/onnx/__main__.py b/optimum/exporters/onnx/__main__.py index 6b7a972a6c1..da018490d69 100644 --- 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