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:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user