add device to ImagesKwargs

huggingface · Jan 16, 2025 · 24087bb · 24087bb
1 parent 3159a25
commit 24087bb
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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):