Update audioldm_train/train/autoencoder.py

audioldm_train/train/autoencoder.py

@@ -13,13 +13,13 @@ import argparse
 import yaml
 import torch
 from pytorch_lightning.strategies.ddp import DDPStrategy
-from audioldm_train.utilities.data.dataset import AudioDataset
+from qa_mdt.audioldm_train.utilities.data.dataset import AudioDataset
 from torch.utils.data import DataLoader
 from pytorch_lightning.loggers import WandbLogger
 from pytorch_lightning import Trainer
-from audioldm_train.modules.latent_encoder.autoencoder import AutoencoderKL
+from qa_mdt.audioldm_train.modules.latent_encoder.autoencoder import AutoencoderKL
 from pytorch_lightning.callbacks import ModelCheckpoint
-from audioldm_train.utilities.tools import get_restore_step
+from qa_mdt.audioldm_train.utilities.tools import get_restore_step
 
 
 def listdir_nohidden(path):