Update internvideo2.py
Browse files- 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 |
|