fix can_generate (#36570)

* fix can_generate

Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

* fix can generate for speecht5 and blip

Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

* fix speecht5 tests

Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

* fix

Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

---------

Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
Co-authored-by: Ilyas Moutawwakil <57442720+IlyasMoutawwakil@users.noreply.github.com>
This commit is contained in:
jiqing-feng
2025-03-17 21:56:18 +08:00
committed by GitHub
parent 9e94801146
commit 2256875a77
4 changed files with 19 additions and 1 deletions

View File

@@ -1076,6 +1076,10 @@ class BarkModelIntegrationTests(unittest.TestCase):
fine_generation_config = BarkFineGenerationConfig(**self.model.generation_config.fine_acoustics_config)
return fine_generation_config
def test_model_can_generate(self):
# Bark has custom generate without inheriting GenerationMixin. This test could prevent regression.
self.assertTrue(self.model.can_generate())
@slow
def test_generate_semantic(self):
input_ids = self.inputs