OpenGVLab
/

InternVideo2_CLIP_S

Model card Files Files and versions

ynhe commited on May 22

Commit

1f9fca1

·

verified ·

1 Parent(s): dfaabf9

Update internvideo2.py

Files changed (1) hide show

internvideo2.py +2 -2

internvideo2.py CHANGED Viewed

@@ -8,8 +8,8 @@ import torch.utils.checkpoint as checkpoint
 from functools import partial
 from einops import rearrange
-from pos_embed import get_3d_sincos_pos_embed, get_2d_sincos_pos_embed, get_1d_sincos_pos_embed, interpolate_pos_embed_internvideo2
-from flash_attention_class import FlashAttention
 from transformers.utils import logging as error_logging

 from functools import partial
 from einops import rearrange
+from .pos_embed import get_3d_sincos_pos_embed, get_2d_sincos_pos_embed, get_1d_sincos_pos_embed, interpolate_pos_embed_internvideo2
+from .flash_attention_class import FlashAttention
 from transformers.utils import logging as error_logging