Update Bark generation configs and tests (#25409)

* update bark generation configs for more coherent parameter

* make style

* update bark hub repo
This commit is contained in:
Yoach Lacombe
2023-08-09 18:28:02 +02:00
committed by GitHub
parent cf84738d2e
commit 704bf595eb
4 changed files with 39 additions and 20 deletions

View File

@@ -26,7 +26,7 @@ from transformers.testing_utils import require_torch, slow
@require_torch
class BarkProcessorTest(unittest.TestCase):
def setUp(self):
self.checkpoint = "ylacombe/bark-small"
self.checkpoint = "suno/bark-small"
self.tmpdirname = tempfile.mkdtemp()
self.voice_preset = "en_speaker_1"
self.input_string = "This is a test string"