Fix repo consistency (#18682)

huggingface · Aug 18, 2022 · 5987c63 · 5987c63
1 parent 76454b0
commit 5987c63
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"}),
             ]
         )