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