Update audioldm_train/modules/latent_diffusion/ddpm.py

audioldm_train/modules/latent_diffusion/ddpm.py

@@ -17,10 +17,10 @@ from functools import partial
 from tqdm import tqdm
 from torchvision.utils import make_grid
 from pytorch_lightning.utilities.rank_zero import rank_zero_only
-from audioldm_train.conditional_models import *
+from qa_mdt.audioldm_train.conditional_models import *
 import datetime
 
-from audioldm_train.utilities.model_util import (
+from qa_mdt.audioldm_train.utilities.model_util import (
     exists,
     default,
     mean_flat,
@@ -28,22 +28,22 @@ from audioldm_train.utilities.model_util import (
     instantiate_from_config,
 )
 
-from audioldm_train.utilities.diffusion_util import (
+from qa_mdt.audioldm_train.utilities.diffusion_util import (
     make_beta_schedule,
     extract_into_tensor,
     noise_like,
 )
 
-from audioldm_train.modules.diffusionmodules.ema import LitEma
-from audioldm_train.modules.diffusionmodules.distributions import (
+from qa_mdt.audioldm_train.modules.diffusionmodules.ema import LitEma
+from qa_mdt.audioldm_train.modules.diffusionmodules.distributions import (
     normal_kl,
     DiagonalGaussianDistribution,
 )
 
 # from audioldm_train.modules.diffusionmodules.transport import 
 
-from audioldm_train.modules.latent_diffusion.ddim import DDIMSampler
-from audioldm_train.modules.latent_diffusion.plms import PLMSSampler
+from qa_mdt.audioldm_train.modules.latent_diffusion.ddim import DDIMSampler
+from qa_mdt.audioldm_train.modules.latent_diffusion.plms import PLMSSampler
 import soundfile as sf
 import os