Update audioldm_train/train/latent_diffusion.py

audioldm_train/train/latent_diffusion.py

@@ -15,7 +15,7 @@ from tqdm import tqdm
 from pytorch_lightning.strategies.ddp import DDPStrategy
 
 
-from audioldm_train.modules.latent_diffusion.ddpm import LatentDiffusion
+from qa_mdt.audioldm_train.modules.latent_diffusion.ddpm import LatentDiffusion
 
 
 from torch.utils.data import WeightedRandomSampler
@@ -25,13 +25,13 @@ from pytorch_lightning.callbacks import ModelCheckpoint
 from pytorch_lightning.loggers import WandbLogger
 
 
-from audioldm_train.utilities.tools import (
+from qa_mdt.audioldm_train.utilities.tools import (
     listdir_nohidden,
     get_restore_step,
     copy_test_subset_data,
 )
 import wandb
-from audioldm_train.utilities.model_util import instantiate_from_config
+from qa_mdt.audioldm_train.utilities.model_util import instantiate_from_config
 import logging
 
 logging.basicConfig(level=logging.WARNING)
@@ -75,7 +75,7 @@ def main(configs, config_yaml_path, exp_group_name, exp_name, perform_validation
 
     #try:
     mos_path = configs["mos_path"]
-    from audioldm_train.utilities.data.hhhh import AudioDataset
+    from qa_mdt.audioldm_train.utilities.data.hhhh import AudioDataset
     dataset = AudioDataset(config=configs, lmdb_path=train_lmdb_path, key_path=train_key_path, mos_path=mos_path)