From c34f5edfd716e02858b09ed88f44c955a7b7b059 Mon Sep 17 00:00:00 2001 From: James Betker Date: Thu, 27 Jan 2022 23:21:44 -0700 Subject: [PATCH] Some renaming --- do_tts.py | 5 ++--- models/{unified_voice.py => autoregressive.py} | 0 models/{lucidrains_dvae.py => dvae.py} | 0 3 files changed, 2 insertions(+), 3 deletions(-) rename models/{unified_voice.py => autoregressive.py} (100%) rename models/{lucidrains_dvae.py => dvae.py} (100%) diff --git a/do_tts.py b/do_tts.py index 3f0e5c9..a508248 100644 --- a/do_tts.py +++ b/do_tts.py @@ -5,14 +5,13 @@ import random import torch import torch.nn.functional as F import torchaudio -import yaml +from models.dvae import DiscreteVAE +from models.autoregressive import UnifiedVoice from tqdm import tqdm from models.arch_util import TorchMelSpectrogram from models.discrete_diffusion_vocoder import DiscreteDiffusionVocoder -from models.lucidrains_dvae import DiscreteVAE from models.text_voice_clip import VoiceCLIP -from models.unified_voice import UnifiedVoice from utils.audio import load_audio from utils.diffusion import SpacedDiffusion, space_timesteps, get_named_beta_schedule from utils.tokenizer import VoiceBpeTokenizer diff --git a/models/unified_voice.py b/models/autoregressive.py similarity index 100% rename from models/unified_voice.py rename to models/autoregressive.py diff --git a/models/lucidrains_dvae.py b/models/dvae.py similarity index 100% rename from models/lucidrains_dvae.py rename to models/dvae.py