riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-03 12:50:06 +06:00
Code Issues Actions 28 Packages Projects Releases Wiki Activity

fix Gemma3nProcessorTest (#39068)

Browse Source 
* fix

* fix

* oups forgot style

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
Co-authored-by: Cyril Vallez <cyril.vallez@gmail.com>
This commit is contained in:
Yih-Dar 2025-06-27 12:28:10 +02:00 committed by GitHub
parent 0d66ef7792
commit 540a10848c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/models/gemma3n/test_processing_gemma3n.py
View File

@ -36,7 +36,7 @@ if is_speech_available():
class Gemma3nProcessorTest(unittest.TestCase):
def setUp(self):
# TODO: update to google?
self.model_id = "Google/gemma-3n-E4B-it"
self.model_id = "hf-internal-testing/namespace-google-repo_name-gemma-3n-E4B-it"
self.tmpdirname = tempfile.mkdtemp(suffix="gemma3n")
self.maxDiff = None
@ -71,6 +71,9 @@ class Gemma3nProcessorTest(unittest.TestCase):
self.assertIsInstance(processor.tokenizer, GemmaTokenizerFast)
self.assertEqual(processor.tokenizer.get_vocab(), tokenizer.get_vocab())
# `disable_grouping` is a new attribute that got added on main while gemma3n was being released - so was
# not part of the saved processor
del processor.feature_extractor.disable_grouping
self.assertIsInstance(processor.feature_extractor, Gemma3nAudioFeatureExtractor)
self.assertEqual(processor.feature_extractor.to_json_string(), feature_extractor.to_json_string())
@ -94,6 +97,9 @@ class Gemma3nProcessorTest(unittest.TestCase):
self.assertEqual(processor.tokenizer.get_vocab(), tokenizer_add_kwargs.get_vocab())
self.assertIsInstance(processor.tokenizer, GemmaTokenizerFast)
# `disable_grouping` is a new attribute that got added on main while gemma3n was being released - so was
# not part of the saved processor
del processor.feature_extractor.disable_grouping
self.assertEqual(processor.feature_extractor.to_json_string(), feature_extractor_add_kwargs.to_json_string())
self.assertIsInstance(processor.feature_extractor, Gemma3nAudioFeatureExtractor)