jadechoghari
commited on
Commit
•
cae8059
1
Parent(s):
7869f0a
Update audioldm_train/modules/audiomae/models_vit.py
Browse files
audioldm_train/modules/audiomae/models_vit.py
CHANGED
@@ -16,7 +16,7 @@ import torch.nn as nn
|
|
16 |
import numpy as np
|
17 |
import timm.models.vision_transformer
|
18 |
from timm.models.vision_transformer import PatchEmbed, Block
|
19 |
-
from audioldm_train.modules.audiomae.util.patch_embed import (
|
20 |
PatchEmbed_new,
|
21 |
PatchEmbed3D_new,
|
22 |
)
|
|
|
16 |
import numpy as np
|
17 |
import timm.models.vision_transformer
|
18 |
from timm.models.vision_transformer import PatchEmbed, Block
|
19 |
+
from qa_mdt.audioldm_train.modules.audiomae.util.patch_embed import (
|
20 |
PatchEmbed_new,
|
21 |
PatchEmbed3D_new,
|
22 |
)
|