diff --git a/src/transformers/processing_utils.py b/src/transformers/processing_utils.py index 611e2aa3f20cd0..b94230c7d4a1da 100644 --- a/src/transformers/processing_utils.py +++ b/src/transformers/processing_utils.py @@ -171,6 +171,8 @@ class methods and docstrings. The channel dimension format for the output image. input_data_format (`ChannelDimension` or `str`, *optional*): The channel dimension format for the input image. + device (`str`, *optional*): + The device to use for processing (e.g. "cpu", "cuda"), only relevant for fast image processing. """ do_resize: Optional[bool] @@ -188,6 +190,7 @@ class methods and docstrings. do_center_crop: Optional[bool] data_format: Optional[ChannelDimension] input_data_format: Optional[Union[str, ChannelDimension]] + device: Optional[str] class VideosKwargs(TypedDict, total=False):