diff --git a/src/transformers/models/levit/configuration_levit.py b/src/transformers/models/levit/configuration_levit.py index a1113d7a7512ce..69032a1faae6ad 100644 --- a/src/transformers/models/levit/configuration_levit.py +++ b/src/transformers/models/levit/configuration_levit.py @@ -137,7 +137,7 @@ class LevitOnnxConfig(OnnxConfig): def inputs(self) -> Mapping[str, Mapping[int, str]]: return OrderedDict( [ - ("pixel_values", {0: "batch", 1: "sequence"}), + ("pixel_values", {0: "batch", 1: "num_channels"}), ] )