Remove outdated VideoInput import in processing_sarashina2_vision.py
Browse files`VideoInput` is not in `transformers.image_utils` anymore. Based on the processor file, sarashina doesn't accept video input. Thus, this PR removes this outdated import.
processing_sarashina2_vision.py
CHANGED
|
@@ -38,7 +38,6 @@ from transformers.image_transforms import (
|
|
| 38 |
from transformers.image_utils import (
|
| 39 |
ChannelDimension,
|
| 40 |
ImageInput,
|
| 41 |
-
VideoInput,
|
| 42 |
get_image_size,
|
| 43 |
infer_channel_dimension_format,
|
| 44 |
is_scaled_image,
|
|
@@ -98,7 +97,7 @@ class GenerationStopper(StoppingCriteria):
|
|
| 98 |
class Sarashina2VisionImageProcessor(Qwen2VLImageProcessor):
|
| 99 |
def _preprocess(
|
| 100 |
self,
|
| 101 |
-
images: Union[ImageInput
|
| 102 |
do_resize: bool = None,
|
| 103 |
resample: Image.Resampling = None,
|
| 104 |
do_rescale: bool = None,
|
|
|
|
| 38 |
from transformers.image_utils import (
|
| 39 |
ChannelDimension,
|
| 40 |
ImageInput,
|
|
|
|
| 41 |
get_image_size,
|
| 42 |
infer_channel_dimension_format,
|
| 43 |
is_scaled_image,
|
|
|
|
| 97 |
class Sarashina2VisionImageProcessor(Qwen2VLImageProcessor):
|
| 98 |
def _preprocess(
|
| 99 |
self,
|
| 100 |
+
images: Union[ImageInput],
|
| 101 |
do_resize: bool = None,
|
| 102 |
resample: Image.Resampling = None,
|
| 103 |
do_rescale: bool = None,
|