Update modeling_dots_ocr_vllm.py
Browse files
modeling_dots_ocr_vllm.py
CHANGED
@@ -28,9 +28,9 @@ from vllm.multimodal.inputs import MultiModalDataDict
|
|
28 |
from vllm.multimodal.parse import ImageSize
|
29 |
from vllm.sequence import IntermediateTensors
|
30 |
|
31 |
-
from
|
32 |
-
from
|
33 |
-
from
|
34 |
|
35 |
|
36 |
class DotsOCRImagePixelInputs(TypedDict):
|
|
|
28 |
from vllm.multimodal.parse import ImageSize
|
29 |
from vllm.sequence import IntermediateTensors
|
30 |
|
31 |
+
from configuration_dots import DotsVisionConfig
|
32 |
+
from configuration_dots import DotsOCRConfig
|
33 |
+
from modeling_dots_vision import DotsVisionTransformer
|
34 |
|
35 |
|
36 |
class DotsOCRImagePixelInputs(TypedDict):
|