From d88a5ca1836b8ed833c308ec02513f7217b76870 Mon Sep 17 00:00:00 2001 From: mrq Date: Thu, 16 May 2024 07:25:33 -0500 Subject: [PATCH] ugh --- vall_e/data.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vall_e/data.py b/vall_e/data.py index c242585..bbbe16b 100755 --- a/vall_e/data.py +++ b/vall_e/data.py @@ -373,7 +373,7 @@ class Dataset(_Dataset): """ prom_length = 0 - trim_length = random.randint(cfg.dataset.prompt_duration_range[0], cfg.dataset.prompt_duration_range[1]) * cfg.dataset.frames_per_second + trim_length = int(random.uniform(cfg.dataset.prompt_duration_range[0], cfg.dataset.prompt_duration_range[1]) * cfg.dataset.frames_per_second) for _ in range(cfg.dataset.max_prompts): path = random.choice(choices) @@ -474,7 +474,7 @@ class Dataset(_Dataset): resps = torch.concat([ resps, qnt ]) task = "tts" - trim_length = random.randint(cfg.dataset.prompt_duration_range[0], cfg.dataset.prompt_duration_range[1]) * cfg.dataset.frames_per_second + trim_length = int(random.uniform(cfg.dataset.prompt_duration_range[0], cfg.dataset.prompt_duration_range[1]) * cfg.dataset.frames_per_second) proms = self.sample_prompts(spkr_name, ignore=path) if random.random() < cfg.dataset.random_utterance else resps