Review cmnt

huggingface · Dec 11, 2023 · e895124 · e895124
1 parent cb3c351
commit e895124
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/optimum/habana/transformers/generation/utils.py b/optimum/habana/transformers/generation/utils.py
@@ -582,9 +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 if generation_config.reduce_recompile is not None else False
+        )
         if model_kwargs["reduce_recompile"]:
             assert generation_config.bucket_size
         if generation_config.reuse_cache: