riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-04 05:10:06 +06:00
Code Issues Actions 20 Packages Projects Releases Wiki Activity

[tests] remove overload for deleted test (test_offloaded_cache_implementation) (#37896)

Browse Source 
* remove overload for deleted tests

* make fixup
This commit is contained in:
Joao Gante 2025-05-27 16:45:15 +01:00 committed by GitHub
parent f85fd90407
commit 0f5a8243c4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 0 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/models/mllama/test_modeling_mllama.py
View File

@ -17,7 +17,6 @@ import unittest
import pytest import pytest
import requests import requests
from parameterized import parameterized
from transformers import ( from transformers import (
AutoProcessor, AutoProcessor,
@ -396,12 +395,6 @@ class MllamaForConditionalGenerationModelTest(ModelTesterMixin, GenerationTester
def test_model_parallelism(self): def test_model_parallelism(self):
pass pass
@parameterized.expand([("offloaded",)])
@pytest.mark.generate
@unittest.skip(reason="Offloaded cache seems to not work with mllama's kv cache type")
def test_offloaded_cache_implementation(self, cache_implementation):
pass
@unittest.skip( @unittest.skip(
reason="Mllama cache type doesn't allow correct check on output `past_key_values` due to `Cache.crop()`" reason="Mllama cache type doesn't allow correct check on output `past_key_values` due to `Cache.crop()`"
) )

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

@ -546,12 +546,6 @@ class WhisperModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMi
def test_generate_with_head_masking(self): def test_generate_with_head_masking(self):
pass pass
@parameterized.expand([("offloaded",)])
@pytest.mark.generate
@unittest.skip(reason="Whisper doesn't work with offloaded cache implementation yet")
def test_offloaded_cache_implementation(self, cache_implementation):
pass
@require_torch_fp16 @require_torch_fp16
def test_generate_fp16(self): def test_generate_fp16(self):
config, input_dict = self.model_tester.prepare_config_and_inputs() config, input_dict = self.model_tester.prepare_config_and_inputs()