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)