Update modeling_dots_vision.py
Browse files- modeling_dots_vision.py +1 -1
modeling_dots_vision.py
CHANGED
@@ -7,7 +7,7 @@ import torch.utils.checkpoint
|
|
7 |
from flash_attn import flash_attn_varlen_func
|
8 |
from torch.nn import LayerNorm
|
9 |
from transformers.modeling_utils import PreTrainedModel
|
10 |
-
from
|
11 |
|
12 |
|
13 |
def rotate_half(x):
|
|
|
7 |
from flash_attn import flash_attn_varlen_func
|
8 |
from torch.nn import LayerNorm
|
9 |
from transformers.modeling_utils import PreTrainedModel
|
10 |
+
from configuration_dots import DotsVisionConfig
|
11 |
|
12 |
|
13 |
def rotate_half(x):
|