From 2c8d512997b4ed77abdf3a763b298ec821d84f93 Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Sun, 29 Aug 2021 05:05:47 -0400
Subject: [PATCH] adjust relative imports

---
 models/tts/ControllableTalkNet/controllable_talknet.py | 10 +++++-----
 models/tts/hifigan/denoiser.py                         |  2 +-
 models/tts/hifigan/stft.py                             |  2 +-
 3 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/models/tts/ControllableTalkNet/controllable_talknet.py b/models/tts/ControllableTalkNet/controllable_talknet.py
index 3f68cc0077..f258054d9f 100644
--- a/models/tts/ControllableTalkNet/controllable_talknet.py
+++ b/models/tts/ControllableTalkNet/controllable_talknet.py
@@ -20,7 +20,7 @@ from nemo.collections.asr.models import EncDecCTCModel
 from nemo.collections.tts.models import TalkNetSpectModel
 from nemo.collections.tts.models import TalkNetPitchModel
 from nemo.collections.tts.models import TalkNetDursModel
-from talknet_singer import TalkNetSingerModel
+from ControllableTalkNet.talknet_singer import TalkNetSingerModel
 import json
 from tqdm import tqdm
 import gdown
@@ -32,10 +32,10 @@ import time
 import uuid
 
 sys.path.append("hifi-gan")
-from env import AttrDict
-from meldataset import mel_spectrogram, MAX_WAV_VALUE
-from models import Generator
-from denoiser import Denoiser
+from hifigan.env import AttrDict
+from hifigan.meldataset import mel_spectrogram, MAX_WAV_VALUE
+from hifigan.models import Generator
+from hifigan.denoiser import Denoiser
 
 app = JupyterDash(__name__)
 DEVICE = "cuda:0"
diff --git a/models/tts/hifigan/denoiser.py b/models/tts/hifigan/denoiser.py
index 4d21d104ce..fe5c4d0ec5 100644
--- a/models/tts/hifigan/denoiser.py
+++ b/models/tts/hifigan/denoiser.py
@@ -1,6 +1,6 @@
 import sys
 import torch
-from stft import STFT
+from hifigan.stft import STFT
 
 
 class Denoiser(torch.nn.Module):
diff --git a/models/tts/hifigan/stft.py b/models/tts/hifigan/stft.py
index edfc44ae8b..16d3810c85 100644
--- a/models/tts/hifigan/stft.py
+++ b/models/tts/hifigan/stft.py
@@ -36,7 +36,7 @@ import torch.nn.functional as F
 from torch.autograd import Variable
 from scipy.signal import get_window
 from librosa.util import pad_center, tiny
-from audio_processing import window_sumsquare
+from hifigan.audio_processing import window_sumsquare
 
 
 class STFT(torch.nn.Module):