Fix repo consistency (huggingface#18682)

oneraghavan · Sep 26, 2022 · 9bda934 · 9bda934
1 parent 79f90b4
commit 9bda934
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"}),
             ]
         )