[ NougatProcessor] Fix the default channel (#26608)

fix
huggingface · Oct 5, 2023 · b4e66d7 · b4e66d7
1 parent 43bfd09
commit b4e66d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/models/nougat/processing_nougat.py b/src/transformers/models/nougat/processing_nougat.py
@@ -61,7 +61,7 @@ def __call__(
         do_normalize: bool = None,
         image_mean: Optional[Union[float, List[float]]] = None,
         image_std: Optional[Union[float, List[float]]] = None,
-        data_format: Optional["ChannelDimension"] = "ChannelDimension.FIRST",  # noqa: F821
+        data_format: Optional["ChannelDimension"] = "channels_first",  # noqa: F821
         input_data_format: Optional[Union[str, "ChannelDimension"]] = None,  # noqa: F821
         text_pair: Optional[Union[TextInput, PreTokenizedInput, List[TextInput], List[PreTokenizedInput]]] = None,
         text_target: Union[TextInput, PreTokenizedInput, List[TextInput], List[PreTokenizedInput]] = None,