fix

Author: xadupre

_unittests/ut_tasks/test_tasks_image_text_to_text.py

@@ -22,6 +22,7 @@ def test_image_text_to_text_idefics(self):
         self.assertEqual(data["task"], "image-text-to-text")
         model, inputs, ds = data["model"], data["inputs"], data["dynamic_shapes"]
         model(**torch_deepcopy(inputs))
+        print("***", self.string_type(data["inputs2"], with_shape=True))
         model(**data["inputs2"])
         with torch_export_patches(patch_transformers=True, verbose=10):
             torch.export.export(

onnx_diagnostic/export/dynamic_shapes.py

@@ -57,7 +57,7 @@ def __init__(
         self.dynamic_shapes = dynamic_shapes
         self.args_names = args_names
         if not self.kwargs and isinstance(self.dynamic_shapes, dict):
-            # This assumes the dicionary for the dynamic shapes is ordered
+            # This assumes the dictionary for the dynamic shapes is ordered
             # the same way the args are. The input names are not known.
             assert len(self.dynamic_shapes) == len(self.args), (
                 f"Length mismatch, kwargs is empty, len(dynamic_shapes)="

onnx_diagnostic/tasks/image_text_to_text.py

@@ -290,7 +290,8 @@ def get_inputs_default(
     )
     if total_sequence_length > 0:
         input_ids[0, 0] = image_token_index
-        input_ids[1, 1] = image_token_index
+        if min(input_ids.shape) > 1:
+            input_ids[1, 1] = image_token_index
     # input_ids[input_ids == image_token_index] = pad_token_id
     token_type_ids = torch.zeros_like(input_ids)
     token_type_ids[input_ids == image_token_index] = 1
@@ -439,9 +440,9 @@ def get_inputs(
             height=height,
             num_channels=num_channels,
             batch_size=3,
-            sequence_length=0,
-            max_sequence_length=0,
-            total_sequence_length=0,
+            sequence_length=1,
+            max_sequence_length=1,
+            total_sequence_length=1,
             n_images=0,
             pad_token_id=pad_token_id,
             image_token_index=image_token_index,