sino commited on
Commit
354c8fa
1 Parent(s): a8381e0

Update modeling_maelm.py

Browse files
Files changed (1) hide show
  1. modeling_maelm.py +4 -4
modeling_maelm.py CHANGED
@@ -5,10 +5,10 @@ from mmcv.cnn.bricks import padding
5
  import torch
6
  from torch import nn, einsum
7
  from typing import Optional, Dict, Tuple
8
- from .src.mae_vit import MAEViT
9
- from .src.htsat import HTSAT_Swin_Transformer, create_htsat_model
10
- from .src.LMdecoder import LMDecoder, LMDecoder_qlora
11
- from .src.vision_transformer import VisionTransformer
12
  from einops import rearrange, repeat
13
  from einops_exts import rearrange_many
14
  import inspect
 
5
  import torch
6
  from torch import nn, einsum
7
  from typing import Optional, Dict, Tuple
8
+ from .mae_vit import MAEViT
9
+ from .htsat import HTSAT_Swin_Transformer, create_htsat_model
10
+ from .LMdecoder import LMDecoder, LMDecoder_qlora
11
+ from .vision_transformer import VisionTransformer
12
  from einops import rearrange, repeat
13
  from einops_exts import rearrange_many
14
  import inspect