style

huggingface · Dec 10, 2023 · cb3c351 · cb3c351
1 parent 837abcd
commit cb3c351
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/optimum/habana/transformers/generation/utils.py b/optimum/habana/transformers/generation/utils.py
@@ -582,7 +582,9 @@ def generate(
             or self._get_generation_mode(generation_config, assistant_model) == GenerationMode.BEAM_SEARCH
         )
         model_kwargs["bucket_size"] = generation_config.bucket_size if generation_config.static_shapes else -1
-        model_kwargs["reduce_recompile"] = (generation_config.reduce_recompile, False)[generation_config.reduce_recompile is None]
+        model_kwargs["reduce_recompile"] = (generation_config.reduce_recompile, False)[
+            generation_config.reduce_recompile is None
+        ]
         if model_kwargs["reduce_recompile"]:
             assert generation_config.bucket_size
         if generation_config.reuse_cache: