Update internvideo2.py

internvideo2.py

@@ -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 .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