fix chronoedit tests on PRs (#13870)

up

Author: sayakpaul

tests/models/transformers/test_models_transformer_chronoedit.py

@@ -40,11 +40,11 @@ def main_input_name(self) -> str:
 
     @property
     def output_shape(self) -> tuple:
-        return (16, 8, 8)
+        return (16, 2, 8, 8)
 
     @property
     def input_shape(self) -> tuple:
-        return (16, 8, 8)
+        return (16, 2, 8, 8)
 
     @property
     def generator(self):