Update optimization.py

optimization.py

@@ -20,7 +20,7 @@ TRANSFORMER_IMAGE_WIDTH_DIM = torch.export.Dim.DYNAMIC
 
 TRANSFORMER_DYNAMIC_SHAPES = {
     'hidden_states': {1: TRANSFORMER_IMAGE_SEQ_LENGTH_DIM},
-    'img_shapes': [(None, TRANSFORMER_IMAGE_HEIGHT_DIM, TRANSFORMER_IMAGE_WIDTH_DIM)]
+    'img_shapes': [[None, TRANSFORMER_IMAGE_HEIGHT_DIM, TRANSFORMER_IMAGE_WIDTH_DIM]]
 }