Update audioldm_train/modules/latent_encoder/autoencoder.py

audioldm_train/modules/latent_encoder/autoencoder.py

@@ -6,20 +6,20 @@ import pytorch_lightning as pl
 import torch.nn.functional as F
 from contextlib import contextmanager
 import numpy as np
-from audioldm_train.modules.diffusionmodules.ema import *
+from qa_mdt.audioldm_train.modules.diffusionmodules.ema import *
 
 from torch.optim.lr_scheduler import LambdaLR
-from audioldm_train.modules.diffusionmodules.model import Encoder, Decoder
-from audioldm_train.modules.diffusionmodules.distributions import (
+from qa_mdt.audioldm_train.modules.diffusionmodules.model import Encoder, Decoder
+from qa_mdt.audioldm_train.modules.diffusionmodules.distributions import (
     DiagonalGaussianDistribution,
 )
 
 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 soundfile as sf
 
-from audioldm_train.utilities.model_util import get_vocoder
-from audioldm_train.utilities.tools import synth_one_sample
+from qa_mdt.audioldm_train.utilities.model_util import get_vocoder
+from qa_mdt.audioldm_train.utilities.tools import synth_one_sample
 import itertools
 
 
@@ -150,7 +150,7 @@ class AutoencoderKL(pl.LightningModule):
         return dec
 
     def decode_to_waveform(self, dec):
-        from audioldm_train.utilities.model_util import vocoder_infer
+        from qa_mdt.audioldm_train.utilities.model_util import vocoder_infer
 
         if self.image_key == "fbank":
             dec = dec.squeeze(1).permute(0, 2, 1)