From e201746eeb3f5be602ae3395df8344f231a5f0d4 Mon Sep 17 00:00:00 2001 From: mrq Date: Thu, 16 Mar 2023 14:19:24 +0000 Subject: [PATCH] added diffusion_model and tokenizer_json as arguments for settings editing --- tortoise/api.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/tortoise/api.py b/tortoise/api.py index 965e133..3aeb5e1 100755 --- a/tortoise/api.py +++ b/tortoise/api.py @@ -565,6 +565,8 @@ class TextToSpeech: num_autoregressive_samples=512, temperature=.8, length_penalty=1, repetition_penalty=2.0, top_p=.8, max_mel_tokens=500, sample_batch_size=None, autoregressive_model=None, + diffusion_model=None, + tokenizer_json=None, # CVVP parameters follow cvvp_amount=.0, # diffusion generation parameters follow @@ -632,6 +634,16 @@ class TextToSpeech: elif autoregressive_model != self.autoregressive_model_path: self.load_autoregressive_model(autoregressive_model) + if diffusion_model is None: + diffusion_model = self.diffusion_model_path + elif diffusion_model != self.diffusion_model_path: + self.load_diffusion_model(diffusion_model) + + if tokenizer_json is None: + tokenizer_json = self.tokenizer_json + elif tokenizer_json != self.tokenizer_json: + self.load_tokenizer_json(tokenizer_json) + text_tokens = torch.IntTensor(self.tokenizer.encode(text)).unsqueeze(0) text_tokens = migrate_to_device( text_tokens, self.device )