Skip to content

GitLab

Unverified Commit e95d433d authored by Joao Gante's avatar Joao Gante Committed by GitHub
Browse files

Generate: add missing `**model_kwargs` in sample tests (#18696)

parent e54a1b49
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
tests/generation/test_generation_utils.py
View file @ e95d433d
...@@ -327,6 +327,7 @@ class GenerationTesterMixin: ...@@ -327,6 +327,7 @@ class GenerationTesterMixin:
remove_invalid_values=True, remove_invalid_values=True,
**logits_warper_kwargs, **logits_warper_kwargs,
**process_kwargs, **process_kwargs,
**model_kwargs,
) )
torch.manual_seed(0) torch.manual_seed(0)
...@@ -361,6 +362,7 @@ class GenerationTesterMixin: ...@@ -361,6 +362,7 @@ class GenerationTesterMixin:
**kwargs, **kwargs,
**model_kwargs, **model_kwargs,
) )
return output_sample, output_generate return output_sample, output_generate
def _beam_search_generate( def _beam_search_generate(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment