fix

Author: xadupre

CHANGELOGS.rst

@@ -4,7 +4,8 @@ Change Logs
 0.7.13
 ++++++
 
-* :pr:`256`: add a set of inputs checking models works for an empty cache on task text-generation
+* :pr:`247`: supports more gemma models with ModelBuilder
+* :pr:`246`: add a set of inputs checking models works for an empty cache on task text-generation
 * :pr:`237`: dummy inputs for google/gemma-3-4b-it
 * :pr:`244`: add a patch to bypass the exception raised when the dynamic dimension is in {0,1}
 

_unittests/ut_tasks/test_tasks.py

@@ -54,18 +54,19 @@ def test_text_generation_empty_cache(self):
         model, inputs = data["model"], data["inputs"]
         self.assertIn("inputs_empty_cache", data)
         empty_inputs = torch_deepcopy(data["inputs_empty_cache"])
-        expected = model(**empty_inputs)
+        model(**torch_deepcopy(empty_inputs))
+        expected = model(**torch_deepcopy(inputs))
         self.assertEqual(
             {"attention_mask", "past_key_values", "input_ids", "position_ids"}, set(inputs)
         )
         with torch_export_patches(patch_transformers=True, verbose=1):
             ep = torch.export.export(
                 model,
                 (),
-                kwargs=inputs,
+                kwargs=torch_deepcopy(inputs),
                 dynamic_shapes=use_dyn_not_str(data["dynamic_shapes"]),
             )
-            got = ep.module()(**inputs)
+            got = ep.module()(**torch_deepcopy(inputs))
             self.assertEqualArrayAny(expected, got)
 
     @hide_stdout()

onnx_diagnostic/torch_models/validate.py

@@ -415,7 +415,8 @@ def validate_model(
     ``orteval10`` increases the verbosity.
 
     .. versionchanged:: 0.7.13
-        *inputs2* not only means a second set of inputs such as ``input_empty_cache``
+        *inputs2* not only means a second set of inputs but many
+        such as ``input_empty_cache``
         which refers to a set of inputs using an empty cache.
     """
     validation_begin = time.perf_counter()