From d8de885311fd63d5e495bd89afd7f65f3e7db700 Mon Sep 17 00:00:00 2001 From: yumoqing Date: Tue, 22 Oct 2024 11:44:09 +0800 Subject: [PATCH] bugfix --- f5tts.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/f5tts.py b/f5tts.py index f961d45..0c97051 100644 --- a/f5tts.py +++ b/f5tts.py @@ -84,10 +84,10 @@ class F5TTS: vocos = Vocos.from_pretrained("charactr/vocos-mel-24khz") self.vocos = vocos - self.F5TTS_model_cfg = dict( - dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4 - ) - self.E2TTS_model_cfg = dict(dim=1024, depth=24, heads=16, ff_mult=4) + self.F5TTS_model_cfg = dict( + dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4 + ) + self.E2TTS_model_cfg = dict(dim=1024, depth=24, heads=16, ff_mult=4) self.model= self.load_model(self.modelname)