SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

VLM generate: tests can't generate image/video tokens (#33623)

Browse Source
This commit is contained in:
Joao Gante
2024-09-20 15:43:27 +01:00
committed by GitHub
parent 653eb40425
commit 2fdb5e74cc
4 changed files with 26 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/models/whisper/test_modeling_whisper.py
View File

@@ -411,9 +411,9 @@ class WhisperModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMi
return False
def _get_logits_processor_kwargs(self, do_sample=False):
def _get_logits_processor_kwargs(self, do_sample=False, config=None):
# Overwritten from `GenerationTesterMixin`, Whisper needs `"temperature": 0.0` to be able to do beam search
logits_processor_kwargs = super()._get_logits_processor_kwargs(do_sample=do_sample)
logits_processor_kwargs = super()._get_logits_processor_kwargs(do_sample=do_sample, config=config)
logits_processor_kwargs["temperature"] = 0.0
return logits_processor_kwargs