Update video_processing_interns1.py
#4
by
jack-zxy
- opened
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
|