From b4e66d7a67f6371f31fe573ac1fd8aa01cb68a93 Mon Sep 17 00:00:00 2001 From: Arthur <48595927+ArthurZucker@users.noreply.github.com> Date: Thu, 5 Oct 2023 09:38:08 +0200 Subject: [PATCH] [ `NougatProcessor`] Fix the default channel (#26608) fix --- src/transformers/models/nougat/processing_nougat.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/models/nougat/processing_nougat.py b/src/transformers/models/nougat/processing_nougat.py index 8e7ba1cb1df68e..b63639e2dd1f7d 100644 --- 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,