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:
remove_invalid_values=True,
**logits_warper_kwargs,
**process_kwargs,
**model_kwargs,
)
torch.manual_seed(0)
......@@ -361,6 +362,7 @@ class GenerationTesterMixin:
**kwargs,
**model_kwargs,
)
return output_sample, output_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