manueldeprada HF Staff commited on
Commit
56e900a
·
verified ·
1 Parent(s): fe87ab3

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. custom_generate/generate.py +1 -0
custom_generate/generate.py CHANGED
@@ -24,6 +24,7 @@ def _constrained_beam_search(
24
  stopping_criteria: StoppingCriteriaList,
25
  generation_config: GenerationConfig,
26
  synced_gpus: bool,
 
27
  **model_kwargs,
28
  ) -> Union[GenerateBeamOutput, torch.LongTensor]:
29
  r"""
 
24
  stopping_criteria: StoppingCriteriaList,
25
  generation_config: GenerationConfig,
26
  synced_gpus: bool,
27
+ streamer: Optional["BaseStreamer"] = None,
28
  **model_kwargs,
29
  ) -> Union[GenerateBeamOutput, torch.LongTensor]:
30
  r"""