New generate function

custom_generate/generate.py

@@ -559,6 +559,8 @@ def generate(
         generation_config is None or generation_config.min_length is None
     )
     generation_config, model_kwargs = model._prepare_generation_config(generation_config, **kwargs)
+    alpha = model_kwargs.pop('alpha')
+    deferral = model_kwargs.pop('deferral')
 
     generation_mode = generation_config.get_generation_mode(assistant_model)
     decoding_method = _speculative_cascades
@@ -721,6 +723,8 @@ def generate(
         logits_processor=prepared_logits_processor,
         stopping_criteria=prepared_stopping_criteria,
         generation_config=generation_config,
+        alpha=alpha,
+        deferral=deferral,
         **generation_mode_kwargs,
         **model_kwargs,
     )