jadechoghari commited on
Commit
7869f0a
1 Parent(s): 4f5251a

Update audioldm_train/modules/audiomae/models_mae.py

Browse files
audioldm_train/modules/audiomae/models_mae.py CHANGED
@@ -16,12 +16,12 @@ import torch
16
  import torch.nn as nn
17
 
18
  from timm.models.vision_transformer import Block
19
- from audioldm_train.modules.audiomae.util.pos_embed import (
20
  get_2d_sincos_pos_embed,
21
  get_2d_sincos_pos_embed_flexible,
22
  get_1d_sincos_pos_embed_from_grid,
23
  )
24
- from audioldm_train.modules.audiomae.util.patch_embed import (
25
  PatchEmbed_new,
26
  PatchEmbed_org,
27
  )
 
16
  import torch.nn as nn
17
 
18
  from timm.models.vision_transformer import Block
19
+ from qa_mdt.audioldm_train.modules.audiomae.util.pos_embed import (
20
  get_2d_sincos_pos_embed,
21
  get_2d_sincos_pos_embed_flexible,
22
  get_1d_sincos_pos_embed_from_grid,
23
  )
24
+ from qa_mdt.audioldm_train.modules.audiomae.util.patch_embed import (
25
  PatchEmbed_new,
26
  PatchEmbed_org,
27
  )