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

[CSM] update model id (#38211)

Browse Source 
* update model id

* codec_model eval

* add processor img

* use ungated repo for processor tests
This commit is contained in:
eustlb
2025-05-27 17:03:55 +02:00
committed by GitHub
parent 07dd6b2495
commit b9f8f863d9
8 changed files with 17 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/csm/test_modeling_csm.py
View File

@@ -417,7 +417,7 @@ class CsmForConditionalGenerationTest(ModelTesterMixin, GenerationTesterMixin, u
class CsmForConditionalGenerationIntegrationTest(unittest.TestCase):
def setUp(self):
# TODO: @eustlb, update with correct sesame's repo
self.model_checkpoint = "eustlb/csm-1b"
self.model_checkpoint = "sesame/csm-1b"
def tearDown(self):
cleanup(torch_device, gc_collect=True)

3
tests/models/csm/test_processor_csm.py
View File

@@ -37,8 +37,7 @@ class CsmProcessorTest(ProcessorTesterMixin, unittest.TestCase):
@classmethod
def setUpClass(cls):
# TODO: @eustlb, change for hf-internal-testing/csm-1b
cls.checkpoint = "eustlb/csm-1b"
cls.checkpoint = "hf-internal-testing/namespace-sesame-repo_name_csm-1b"
processor = CsmProcessor.from_pretrained(cls.checkpoint)
cls.audio_token = processor.audio_token
cls.audio_token_id = processor.audio_token_id