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 |
)
|