ynhe commited on
Commit
1f9fca1
·
verified ·
1 Parent(s): dfaabf9

Update internvideo2.py

Browse files
Files changed (1) hide show
  1. internvideo2.py +2 -2
internvideo2.py CHANGED
@@ -8,8 +8,8 @@ import torch.utils.checkpoint as checkpoint
8
  from functools import partial
9
  from einops import rearrange
10
 
11
- from pos_embed import get_3d_sincos_pos_embed, get_2d_sincos_pos_embed, get_1d_sincos_pos_embed, interpolate_pos_embed_internvideo2
12
- from flash_attention_class import FlashAttention
13
 
14
  from transformers.utils import logging as error_logging
15
 
 
8
  from functools import partial
9
  from einops import rearrange
10
 
11
+ from .pos_embed import get_3d_sincos_pos_embed, get_2d_sincos_pos_embed, get_1d_sincos_pos_embed, interpolate_pos_embed_internvideo2
12
+ from .flash_attention_class import FlashAttention
13
 
14
  from transformers.utils import logging as error_logging
15