From 30abf0b637271643605317a50cdc8b55dd73bb21 Mon Sep 17 00:00:00 2001 From: Cemberk Date: Wed, 19 Feb 2025 20:30:28 +0000 Subject: [PATCH] gfx90a update --- tests/models/qwen2_5_vl/test_modeling_qwen2_5_vl.py | 2 +- tests/models/whisper/test_modeling_whisper.py | 5 +++++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/tests/models/qwen2_5_vl/test_modeling_qwen2_5_vl.py b/tests/models/qwen2_5_vl/test_modeling_qwen2_5_vl.py index 14c0aef3a009..11adedc5ceff 100644 --- a/tests/models/qwen2_5_vl/test_modeling_qwen2_5_vl.py +++ b/tests/models/qwen2_5_vl/test_modeling_qwen2_5_vl.py @@ -237,7 +237,7 @@ class Qwen2_5_VLModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.Test def test_generate_continue_from_past_key_values(self): super().test_generate_continue_from_past_key_values() - @skipIfRocm(arch=['gfx942','gfx1201','gfx1200']) + @skipIfRocm(arch=['gfx942','gfx1201','gfx1200','gfx90a']) def test_generate_with_static_cache(self): super().test_generate_with_static_cache() diff --git a/tests/models/whisper/test_modeling_whisper.py b/tests/models/whisper/test_modeling_whisper.py index 80c4025c259c..2c47485ea042 100644 --- a/tests/models/whisper/test_modeling_whisper.py +++ b/tests/models/whisper/test_modeling_whisper.py @@ -43,6 +43,7 @@ require_torchaudio, slow, torch_device, + skipIfRocm, ) from transformers.utils import cached_property, is_flax_available, is_torch_available, is_torchaudio_available from transformers.utils.import_utils import is_datasets_available @@ -395,6 +396,10 @@ class WhisperModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMi # `0.5` is for `test_disk_offload` (which also works for `test_model_parallelism`) model_split_percents = [0.5, 0.8, 0.9] + @skipIfRocm(arch='gfx90a', os_name='ubuntu', os_version='24.04') + def test_longform_generate_multi_batch_cond_prev(self): + super().test_longform_generate_multi_batch_cond_prev() + # TODO: Fix the failed tests def is_pipeline_test_to_skip( self,