kkokkie2360 commited on
Commit
4c5a985
·
verified ·
1 Parent(s): 021aa10

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.

Files changed (1) hide show
  1. processing_sarashina2_vision.py +1 -2
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, VideoInput],
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,