ops3

Author: manueldeprada

src/transformers/generation/utils.py

@@ -1494,7 +1494,7 @@ def compute_transition_scores(
         return transition_scores
 
     def _validate_assistant(self, **gen_mode_kwargs):
-        if assistant_model := gen_mode_kwargs.get("assistant_model") is None:
+        if (assistant_model := gen_mode_kwargs.get("assistant_model")) is None:
             return
         assistant_tokenizer = gen_mode_kwargs.get("assistant_tokenizer")
 
@@ -2157,7 +2157,7 @@ def _get_deprecated_gen_repo(
             )
         return repo
 
-    def _get_mode_processor_kwargs(self, custom_generate, **kwargs) -> tuple[dict[str, Any], dict[str, Any]]:
+    def _get_mode_processor_kwargs(self, custom_generate, kwargs) -> tuple[dict[str, Any], dict[str, Any]]:
         """
         Extracts and returns the generation mode and logit processor related keyword arguments from the provided kwargs.
         """
@@ -2294,7 +2294,7 @@ def generate(
             return custom_generate_function(model=self, **generate_arguments)
 
         # 1. Handle kwargs, `generation_config`, validate them and obtain generation mode
-        gen_mode_kwargs, logits_processor_kwargs = self._get_mode_processor_kwargs(custom_generate, **kwargs)
+        gen_mode_kwargs, logits_processor_kwargs = self._get_mode_processor_kwargs(custom_generate, kwargs)
 
         generation_config, model_kwargs = self._prepare_generation_config(
             generation_config, use_model_defaults, **kwargs