riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 18:22:34 +06:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Fix flaky test vision encoder-decoder generate (#28923)

Browse Source
This commit is contained in:
Raushan Turganbay 2024-02-14 20:40:57 +05:00 committed by GitHub
parent 0507e69d34
commit 354775bc57
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/models/vision_encoder_decoder/test_modeling_vision_encoder_decoder.py
View File

@ -23,7 +23,6 @@ from packaging import version
from transformers import DonutProcessor, NougatProcessor, TrOCRProcessor from transformers import DonutProcessor, NougatProcessor, TrOCRProcessor
from transformers.testing_utils import ( from transformers.testing_utils import (
is_flaky,
require_levenshtein, require_levenshtein,
require_nltk, require_nltk,
require_sentencepiece, require_sentencepiece,
@ -286,6 +285,8 @@ class EncoderDecoderMixin:
enc_dec_model.config.eos_token_id = None enc_dec_model.config.eos_token_id = None
if hasattr(enc_dec_model.config, "decoder") and hasattr(enc_dec_model.config.decoder, "eos_token_id"): if hasattr(enc_dec_model.config, "decoder") and hasattr(enc_dec_model.config.decoder, "eos_token_id"):
enc_dec_model.config.decoder.eos_token_id = None enc_dec_model.config.decoder.eos_token_id = None
if hasattr(enc_dec_model.generation_config, "eos_token_id"):
enc_dec_model.generation_config.eos_token_id = None
enc_dec_model.to(torch_device) enc_dec_model.to(torch_device)
inputs = pixel_values inputs = pixel_values
@ -324,10 +325,6 @@ class EncoderDecoderMixin:
input_ids_dict = self.prepare_config_and_inputs() input_ids_dict = self.prepare_config_and_inputs()
self.check_encoder_decoder_model_output_attentions(**input_ids_dict) self.check_encoder_decoder_model_output_attentions(**input_ids_dict)
# FIXME @gante: flaky test
@is_flaky(
description="Fails on distributed runs e.g.: https://app.circleci.com/pipelines/github/huggingface/transformers/83611/workflows/666b01c9-1be8-4daa-b85d-189e670fc168/jobs/1078635/tests#failed-test-0"
)
def test_encoder_decoder_model_generate(self): def test_encoder_decoder_model_generate(self):
input_ids_dict = self.prepare_config_and_inputs() input_ids_dict = self.prepare_config_and_inputs()
self.check_encoder_decoder_model_generate(**input_ids_dict) self.check_encoder_decoder_model_generate(**input_ids_dict)