Update video_processing_interns1.py (#6)
Browse files- Update video_processing_interns1.py (af524d72b145ecbfd8a393053e98531377559eca)
Co-authored-by: Zhou Xinyu <[email protected]>
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
|