Fix: relate path package

src/split_audio/main.py

@@ -6,9 +6,9 @@ import phonemizer
 import yaml
 from split_audio.models import load_ASR_models, load_F0_models, build_model
 from split_audio.utils import mask_from_lens, maximum_path
-from utils import length_to_mask, recursive_munch
-from plbert.plbert import load_plbert
-from text_utils import TextCleaner
+from split_audio.utils import length_to_mask, recursive_munch
+from split_audio.plbert.plbert import load_plbert
+from split_audio.text_utils import TextCleaner
 import librosa
 import numpy as np
 import torchaudio

src/split_audio/models.py

@@ -17,11 +17,11 @@ from torch.nn.utils import weight_norm, remove_weight_norm, spectral_norm
 from split_audio.asr.model import ASRCNN
 from split_audio.jdc.model import JDCNet
 
-from Modules.diffusion.sampler import KDiffusion, LogNormalDistribution
-from Modules.diffusion.modules import Transformer1d, StyleTransformer1d
-from Modules.diffusion.diffusion import AudioDiffusionConditional
+from split_audio.Modules.diffusion.sampler import KDiffusion, LogNormalDistribution
+from split_audio.Modules.diffusion.modules import Transformer1d, StyleTransformer1d
+from split_audio.Modules.diffusion.diffusion import AudioDiffusionConditional
 
-from Modules.discriminators import MultiPeriodDiscriminator, MultiResSpecDiscriminator, WavLMDiscriminator
+from split_audio.Modules.discriminators import MultiPeriodDiscriminator, MultiResSpecDiscriminator, WavLMDiscriminator
 
 from munch import Munch
 import yaml