new changes in caches

Author: xadupre

_unittests/ut_torch_models/test_tiny_llms.py

@@ -37,7 +37,6 @@ def test_tiny_llm_export_dynamic(self):
                 dynamic_shapes=use_dyn_not_str(data["dynamic_shapes"]),
             )
             got = ep.module()(**inputs)
-            print(ep)
             self.assertEqualArrayAny(expected, got)
 
     @requires_transformers("4.52")

onnx_diagnostic/helpers/cache_helper.py

@@ -265,6 +265,9 @@ def __init__(self):
             self.num_attention_heads = key_value_pairs[0][0].shape[1]
             self.num_hidden_layers = len(key_value_pairs)
 
+        def get_text_config(self):
+            return self
+
     assert max_cache_len is not None, (
         f"max_cache_len={max_cache_len} cannot be setup "
         f"automatically yet from shape {key_value_pairs[0][0].shape}"