Update audioldm_train/modules/diffusionmodules/openaimodel.py

audioldm_train/modules/diffusionmodules/openaimodel.py

@@ -8,7 +8,7 @@ import torch as th
 import torch.nn as nn
 import torch.nn.functional as F
 
-from audioldm_train.utilities.diffusion_util import (
+from qa_mdt.audioldm_train.utilities.diffusion_util import (
     checkpoint,
     conv_nd,
     linear,
@@ -17,7 +17,7 @@ from audioldm_train.utilities.diffusion_util import (
     normalization,
     timestep_embedding,
 )
-from audioldm_train.modules.diffusionmodules.attention import SpatialTransformer
+from qa_mdt.audioldm_train.modules.diffusionmodules.attention import SpatialTransformer
 
 
 # dummy replace