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

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


Co-authored-by: Tomer Ronen <tomer-nv@users.noreply.huggingface.co>

modeling_decilm.py

@@ -802,10 +802,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