jack-zxy commited on
Commit
12a7333
·
verified ·
1 Parent(s): 206cd5f

Update video_processing_interns1.py

Browse files

Mainly to resolve the following bug:
https://github.com/vllm-project/vllm/issues/25451

The latest transformers will pass extra parameters into the video processor, while the current InternS1 processor lacks `**kwargs`.
Therefore, this PR adds `**kwargs` to allow extra inputs.

Files changed (1) hide show
  1. video_processing_interns1.py +1 -0
video_processing_interns1.py CHANGED
@@ -148,6 +148,7 @@ class InternS1VideoProcessor(BaseVideoProcessor):
148
  initial_shift: Optional[Union[bool, float, int]] = None,
149
  return_tensors: Optional[Union[str, TensorType]] = None,
150
  device: Optional["torch.Tensor"] = None,
 
151
  ) -> BatchFeature:
152
  if do_sample_frames:
153
  # Sample video frames
 
148
  initial_shift: Optional[Union[bool, float, int]] = None,
149
  return_tensors: Optional[Union[str, TensorType]] = None,
150
  device: Optional["torch.Tensor"] = None,
151
+ **kwargs
152
  ) -> BatchFeature:
153
  if do_sample_frames:
154
  # Sample video frames