_prepare_generation_config bugfix (failed due to version update in transformers) (#25)

- _prepare_generation_config bugfix (failed due to version update in transformers) (aabac526c6963d3107659c16ef56c50408e96b65)


Co-authored-by: Ido Shahaf <[email protected]>

modeling_decilm.py

@@ -833,10 +833,13 @@ class DeciLMPreTrainedModel(PreTrainedModel):
                 module.weight.data[module.padding_idx].zero_()
 
     def _prepare_generation_config(
-            self, generation_config: Optional[GenerationConfig], **kwargs: dict
+            self,
+            generation_config: Optional[GenerationConfig],
+            *args,
+            **kwargs,
     ) -> tuple[GenerationConfig, dict]:
         # DeciLM-specific code
-        generation_config, model_kwargs = super()._prepare_generation_config(generation_config, **kwargs)
+        generation_config, model_kwargs = super()._prepare_generation_config(generation_config, *args, **kwargs)
         generation_config.cache_implementation = "variable"
         NEED_SETUP_CACHE_CLASSES_MAPPING["variable"] = VariableCache
         return generation_config, model_kwargs