onnx-diagnostic 0.8.8__py3-none-any.whl → 0.8.10__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,13 +1,7 @@
1
1
  import itertools
2
2
  from typing import Any, Callable, List, Set, Tuple
3
3
  import torch
4
- from transformers.cache_utils import (
5
- Cache,
6
- DynamicCache,
7
- EncoderDecoderCache,
8
- HybridCache,
9
- StaticCache,
10
- )
4
+ from transformers.cache_utils import Cache, DynamicCache, EncoderDecoderCache, StaticCache
11
5
 
12
6
  try:
13
7
  from transformers.cache_utils import SlidingWindowCache
@@ -15,18 +9,17 @@ except ImportError:
15
9
  SlidingWindowCache = None
16
10
 
17
11
 
12
+ try:
13
+ from transformers.cache_utils import HybridCache
14
+ except ImportError:
15
+ HybridCache = None
16
+
18
17
  try:
19
18
  from transformers.models.mamba.modeling_mamba import MambaCache
20
19
  except ImportError:
21
20
  from transformers.cache_utils import MambaCache
22
21
  from transformers.modeling_outputs import BaseModelOutput
23
- from ...helpers.cache_helper import (
24
- make_dynamic_cache,
25
- make_hybrid_cache,
26
- make_sliding_window_cache,
27
- make_static_cache,
28
- CacheKeyValue,
29
- )
22
+ from ...helpers.cache_helper import make_dynamic_cache, make_static_cache, CacheKeyValue
30
23
  from . import make_serialization_function_for_dataclass
31
24
 
32
25
 
@@ -78,6 +71,14 @@ def flatten_dynamic_cache(
78
71
  dynamic_cache: DynamicCache,
79
72
  ) -> Tuple[List[Any], torch.utils._pytree.Context]:
80
73
  """Serializes a :class:`transformers.cache_utils.DynamicCache` with python objects."""
74
+ assert (
75
+ not hasattr(dynamic_cache, "layers")
76
+ or not dynamic_cache.layers
77
+ or all(lay.__class__.__name__ == "DynamicLayer" for lay in dynamic_cache.layers)
78
+ ), (
79
+ f"The serialization does not work yet on other layers "
80
+ f"than DynamicLayer, but layers={[lay.__class__ for lay in dynamic_cache.layers]}"
81
+ )
81
82
  return _flatten_key_value_cache(dynamic_cache)
82
83
 
83
84
 
@@ -85,6 +86,14 @@ def flatten_with_keys_dynamic_cache(
85
86
  dynamic_cache: DynamicCache,
86
87
  ) -> Tuple[List[Tuple[torch.utils._pytree.KeyEntry, Any]], torch.utils._pytree.Context]:
87
88
  """Serializes a :class:`transformers.cache_utils.DynamicCache` with python objects."""
89
+ assert (
90
+ not hasattr(dynamic_cache, "layers")
91
+ or not dynamic_cache.layers
92
+ or all(lay.__class__.__name__ == "DynamicLayer" for lay in dynamic_cache.layers)
93
+ ), (
94
+ f"The serialization does not work yet on other layers "
95
+ f"than DynamicLayer, but layers={[lay.__class__ for lay in dynamic_cache.layers]}"
96
+ )
88
97
  return _flatten_with_keys_cache(dynamic_cache)
89
98
 
90
99
 
@@ -99,26 +108,27 @@ def unflatten_dynamic_cache(
99
108
  # HybridCache
100
109
  #############
101
110
 
111
+ if HybridCache:
102
112
 
103
- def flatten_hybrid_cache(
104
- cache: HybridCache,
105
- ) -> Tuple[List[Any], torch.utils._pytree.Context]:
106
- """Serializes a :class:`transformers.cache_utils.HybridCache` with python objects."""
107
- return _flatten_key_value_cache(cache)
108
-
113
+ def flatten_hybrid_cache(
114
+ cache: HybridCache,
115
+ ) -> Tuple[List[Any], torch.utils._pytree.Context]:
116
+ """Serializes a :class:`transformers.cache_utils.HybridCache` with python objects."""
117
+ return _flatten_key_value_cache(cache)
109
118
 
110
- def flatten_with_keys_hybrid_cache(
111
- cache: HybridCache,
112
- ) -> Tuple[List[Tuple[torch.utils._pytree.KeyEntry, Any]], torch.utils._pytree.Context]:
113
- """Serializes a :class:`transformers.cache_utils.HybridCache` with python objects."""
114
- return _flatten_with_keys_cache(cache)
119
+ def flatten_with_keys_hybrid_cache(
120
+ cache: HybridCache,
121
+ ) -> Tuple[List[Tuple[torch.utils._pytree.KeyEntry, Any]], torch.utils._pytree.Context]:
122
+ """Serializes a :class:`transformers.cache_utils.HybridCache` with python objects."""
123
+ return _flatten_with_keys_cache(cache)
115
124
 
125
+ def unflatten_hybrid_cache(
126
+ values: List[Any], context: torch.utils._pytree.Context, output_type=None
127
+ ) -> HybridCache:
128
+ """Restores a :class:`transformers.cache_utils.HybridCache` from python objects."""
129
+ from ...helpers.cache_helper import make_hybrid_cache
116
130
 
117
- def unflatten_hybrid_cache(
118
- values: List[Any], context: torch.utils._pytree.Context, output_type=None
119
- ) -> HybridCache:
120
- """Restores a :class:`transformers.cache_utils.HybridCache` from python objects."""
121
- return _unflatten_cache(make_hybrid_cache, values, context, output_type=output_type)
131
+ return _unflatten_cache(make_hybrid_cache, values, context, output_type=output_type)
122
132
 
123
133
 
124
134
  #############
@@ -190,6 +200,8 @@ if SlidingWindowCache:
190
200
  Restores a :class:`transformers.cache_utils.SlidingWindowCache`
191
201
  from python objects.
192
202
  """
203
+ from ...helpers.cache_helper import make_sliding_window_cache
204
+
193
205
  return _unflatten_cache(
194
206
  make_sliding_window_cache, values, context, output_type=output_type
195
207
  )
@@ -143,6 +143,7 @@ def _ccached_microsoft_phi2():
143
143
  "transformers_version": "4.51.0.dev0",
144
144
  "use_cache": true,
145
145
  "vocab_size": 51200,
146
+ "pad_token_id": 0,
146
147
  }
147
148
  )
148
149
 
@@ -53,6 +53,7 @@ def get_phi2(
53
53
  "transformers_version": "4.37.0",
54
54
  "use_cache": True,
55
55
  "vocab_size": 51200,
56
+ "pad_token_id": 0,
56
57
  }
57
58
  config.update(**kwargs)
58
59
  conf = transformers.PhiConfig(**config)
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: onnx-diagnostic
3
- Version: 0.8.8
3
+ Version: 0.8.10
4
4
  Summary: Tools to help converting pytorch models into ONNX.
5
5
  Home-page: https://github.com/sdpython/onnx-diagnostic
6
6
  Author: Xavier Dupré
@@ -90,7 +90,7 @@ Enlightening Examples
90
90
 
91
91
  * `Export microsoft/phi-2
92
92
  <https://sdpython.github.io/doc/onnx-diagnostic/dev/auto_examples/plot_export_tiny_phi2.html>`_
93
- * `Export a model through method generate (with Tiny-LLM)
93
+ * `Export a LLM through method generate (with Tiny-LLM)
94
94
  <https://sdpython.github.io/doc/onnx-diagnostic/dev/auto_examples/plot_export_tiny_llm_method_generate.html>`_
95
95
 
96
96
  **Torch Export**
@@ -1,42 +1,42 @@
1
- onnx_diagnostic/__init__.py,sha256=Is5g4yyLoruIq2F94HTISS_8sTf8w9AHzf6qlhhytZ0,173
1
+ onnx_diagnostic/__init__.py,sha256=hheb1Xv21S-ukwOrWVc48HZ9f_jgwqg27V4HdlvfuZQ,174
2
2
  onnx_diagnostic/__main__.py,sha256=YmyV_Aq_ianDlHyKLHMa6h8YK3ZmFPpLVHLKjM91aCk,79
3
3
  onnx_diagnostic/_command_lines_parser.py,sha256=g_udwHBHmY6X_d41Qby_DqMpEHL1p9GfUhJGBCihl8c,57784
4
4
  onnx_diagnostic/api.py,sha256=BhCl_yCd78N7TlVtPOHjeYv1QBEy39TjZ647rcHqLh0,345
5
- onnx_diagnostic/doc.py,sha256=t3RELgfooYnVMAi0JSpggWkQEgUsREz8NmRvn0TnLI8,2829
5
+ onnx_diagnostic/doc.py,sha256=-w2qaes0G0TM0PQFTzBVcJPh0r_IR6Vd2j2fj9iV65Y,10478
6
6
  onnx_diagnostic/ext_test_case.py,sha256=A6BkrRm-QbvM8A-qRRMLt9o9ZO6wMXE9jrotggjpGfE,50460
7
7
  onnx_diagnostic/ci_models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
8
8
  onnx_diagnostic/ci_models/ci_helpers.py,sha256=lblOF7z2kLcCRAwMOdqp-Tz1EL1oBywHfVokhqiTQRg,15592
9
9
  onnx_diagnostic/ci_models/export_phi4_mm.py,sha256=DH225jXhOG_S8KZE9Kn7FGNE4VqFyc9ZDnG-qxZn-hk,41668
10
10
  onnx_diagnostic/ci_models/export_qwen25_vl.py,sha256=_rYPr8PPraWizr2MPcGuYjrJ55ilJOyKl8kg0wq4L90,20405
11
11
  onnx_diagnostic/export/__init__.py,sha256=yEIoWiOeTwBsDhyYt2fTKuhtA0Ya1J9u9ZzMTOTWaWs,101
12
- onnx_diagnostic/export/api.py,sha256=V-2IhcxfMRFIh-CBb3c7QUUTcd_11FjRLHXLhIOy4Fk,22111
12
+ onnx_diagnostic/export/api.py,sha256=7CL_IfVERPKTvKSkFfxlfXfVqUDuYAmDykCEW5CktMo,43202
13
13
  onnx_diagnostic/export/cf_simple_loop_for.py,sha256=OHPGQc9AC-0TBtCYpP6cm-iHP9gmNt8WYRrPlO9ewlc,21158
14
14
  onnx_diagnostic/export/control_flow_onnx.py,sha256=izGlctqQANrHzSxPMbT7hoauNbnIBdx6hb8ry7HtVmM,18263
15
- onnx_diagnostic/export/dynamic_shapes.py,sha256=z6BgCAHyrB6SLcCdjy5CMr8lOy-dm-CUQgJc_GyQVTw,44138
15
+ onnx_diagnostic/export/dynamic_shapes.py,sha256=_3xnWXa4n7fTkAT0NSLyLkcEJqm896wHKA8ilEyxvm0,44746
16
16
  onnx_diagnostic/export/onnx_plug.py,sha256=U13fL0BjnhMzcDGxaAOqM4TQte5Z4zKDg4ESS0iktjM,22704
17
- onnx_diagnostic/export/shape_helper.py,sha256=L0RKUufLQDM5PqDMYk-9L3YZMnv4Zpz2t18KL_37RUU,11341
17
+ onnx_diagnostic/export/shape_helper.py,sha256=K3y8-vFXYGg5R1tgeVOm_RjCZ8-yyrvYkQ4b3ILM5H4,10990
18
18
  onnx_diagnostic/export/validate.py,sha256=_PGUql2DJhIgGKo0WjTGUc5AgsZUx8fEs00MePy-w98,6043
19
19
  onnx_diagnostic/helpers/__init__.py,sha256=GJ2GT7cgnlIveVUwMZhuvUwidbTJaKv8CsSIOpZDsJg,83
20
20
  onnx_diagnostic/helpers/_log_helper.py,sha256=OTwQH0OIxs9B6nrSvR7MoxMimSw_8mU0mj133NvLk5o,16832
21
21
  onnx_diagnostic/helpers/args_helper.py,sha256=SRWnqC7EENg09RZlA50B_PcdiIhdbgA4C3ACfzl5nMs,4419
22
22
  onnx_diagnostic/helpers/bench_run.py,sha256=Vvzb7Wy0baIT5O0dx4RKQTx-5V08PiHxPJh6XPkY-lU,16544
23
- onnx_diagnostic/helpers/cache_helper.py,sha256=RsPuYRQ_RiA4I-J2loD-gXnmLsvxskp_nqv_C9hQlrU,28114
23
+ onnx_diagnostic/helpers/cache_helper.py,sha256=aKy566IGuNTZI14Fk3BlByxPYzuU-401eiXY8GN_UWE,31895
24
24
  onnx_diagnostic/helpers/config_helper.py,sha256=cWRETgFhZ7tayIZPnMqF8BF5AvTU64G2BMqyzgO7lzs,5670
25
25
  onnx_diagnostic/helpers/doc_helper.py,sha256=pl5MZd3_FaE8BqQnqoBuSBxoNCFcd2OJd3eITUSku5c,5897
26
26
  onnx_diagnostic/helpers/dot_helper.py,sha256=hwgTJsbsUv0qq7euyPDnc1NsBZDGOwv32JXSZxIHJkE,8118
27
27
  onnx_diagnostic/helpers/fake_tensor_helper.py,sha256=59046wDIw84or6PJxLaa2CFqaWT7Y3mpYr-BB2shcBE,12027
28
28
  onnx_diagnostic/helpers/graph_helper.py,sha256=hevQT5a7_QuriVPQcbT5qe18n99Doyl5h3-qshx1-uk,14093
29
- onnx_diagnostic/helpers/helper.py,sha256=Fa5wB0CAva2d82RDjCoJxbf1AafYWPZEHKlyz7PZLg0,66505
29
+ onnx_diagnostic/helpers/helper.py,sha256=7rBu2mQVxQxXrW1YuK433Lp68lvgNOhEqUNkuv9GxJw,66454
30
30
  onnx_diagnostic/helpers/log_helper.py,sha256=3mWQd-nLKCctKZt9N8SpoWgLC8O7YdNQ2pfW5QXYWDQ,93232
31
31
  onnx_diagnostic/helpers/memory_peak.py,sha256=M3m4_thWFIwP5HytbJYEqaijXIv5v5BW_vlcJowIYI4,6434
32
32
  onnx_diagnostic/helpers/mini_onnx_builder.py,sha256=jR2lkRZEQ0N30H0FqeBwaxJd_w_6kyxFagrnulqFjhE,23883
33
33
  onnx_diagnostic/helpers/model_builder_helper.py,sha256=qKIq4Naqq03gk6NfqXLQjSDiKL5FFNc1AEyVX0R8GmA,18540
34
- onnx_diagnostic/helpers/onnx_helper.py,sha256=MshvqMSTNUUZIpkkRYGDymdW2t2KtB2BgYtOPHIDwvQ,57508
34
+ onnx_diagnostic/helpers/onnx_helper.py,sha256=f5iyzNIXWS90COkERnEb8txafH19CRYWKdKVY_lF05Q,57801
35
35
  onnx_diagnostic/helpers/optim_helper.py,sha256=0NiYRwV9GLTub4SEny0dqEhLcajRjEhcgkeBDVr9bGQ,4424
36
36
  onnx_diagnostic/helpers/ort_session.py,sha256=Y6iJojJcQtPesy9dRaaxRSeRTLXk0Pdh6-_fzdiPDm0,30779
37
- onnx_diagnostic/helpers/rt_helper.py,sha256=OOxHSCKZup2u7zTvVJxPkRHb4jQZ03KpkiDGrfwibMM,38135
37
+ onnx_diagnostic/helpers/rt_helper.py,sha256=YhVB7v5iR_UT9IqCeqyg58IEIx18f_TnCO6u6GhUciQ,38711
38
38
  onnx_diagnostic/helpers/torch_fx_graph_helper.py,sha256=7xFe4svdbr4gV3OTNcx8eJejjDyHAv4hD_RNNKSxL0c,6571
39
- onnx_diagnostic/helpers/torch_helper.py,sha256=ADV81WWCskQ-jmzn1GI_LU8GqBtufCHS3oeLG-3Uw0E,38954
39
+ onnx_diagnostic/helpers/torch_helper.py,sha256=H_rNXq4G78-45pJ2vsO0b_VxYVpSp6DVrOKaal6-VXo,39346
40
40
  onnx_diagnostic/reference/__init__.py,sha256=rLZsxOlnb7-81F2CzepGnZLejaROg4JvgFaGR9FwVQA,208
41
41
  onnx_diagnostic/reference/evaluator.py,sha256=RzNzjFDeMe-4X51Tb22N6aagazY5ktNq-mRmPcfY5EU,8848
42
42
  onnx_diagnostic/reference/ort_evaluator.py,sha256=q7Dn0yC3LPadlfRnhiRzVn32k9ma_IivdYyhyecgNgc,33930
@@ -89,7 +89,7 @@ onnx_diagnostic/tasks/automatic_speech_recognition.py,sha256=aMufLDGW005f7aLMZ9a
89
89
  onnx_diagnostic/tasks/feature_extraction.py,sha256=IS9z9fPNE0hhGUebBfmNZl0twdXobMc7MFKpQB9qZI0,5388
90
90
  onnx_diagnostic/tasks/fill_mask.py,sha256=5Gt6zlj0p6vuifox7Wmj-TpHXJvPS0CEH8evgdBHDNA,2640
91
91
  onnx_diagnostic/tasks/image_classification.py,sha256=nLpBBB1Gkog3Fk6pu2waiHcuQr4ILPptc9FhQ-pn460,4682
92
- onnx_diagnostic/tasks/image_text_to_text.py,sha256=A3mg3t7zh2Z_kAJ4cQBrW9x6O1knMPjL_LwMlH78m-k,22142
92
+ onnx_diagnostic/tasks/image_text_to_text.py,sha256=LFHngpedt4Ws2lj1v0QIqmbTvpV_IbOE05M3o2GRPfs,22289
93
93
  onnx_diagnostic/tasks/image_to_video.py,sha256=SoF2cVIJr6P30Abp-FCuixFDh5RvTuNEOL36QthGY6U,3860
94
94
  onnx_diagnostic/tasks/mask_generation.py,sha256=fjdD3rd-O-mFL0hQy3la3JXKth_0bH2HL7Eelq-3Dbs,5057
95
95
  onnx_diagnostic/tasks/mixture_of_expert.py,sha256=al4tk1BrHidtRiHlAaiflWiJaAte0d5M8WcBioANG9k,2808
@@ -98,20 +98,20 @@ onnx_diagnostic/tasks/sentence_similarity.py,sha256=vPqNZgAnIvY0rKWPUTs0IlU3RFQD
98
98
  onnx_diagnostic/tasks/summarization.py,sha256=AyDUHLjEymn4waIFf_ZgLAUJT6xqiGFKdaYAikK3wVA,5382
99
99
  onnx_diagnostic/tasks/text2text_generation.py,sha256=E-H5_wZX-RjExpM65-B61eaNx_lJVCCOKo5AN7FnYzc,9873
100
100
  onnx_diagnostic/tasks/text_classification.py,sha256=CGc72SpXFzTUyzAHEMPgyy_s187DaYGsRdrosxG80_Q,2711
101
- onnx_diagnostic/tasks/text_generation.py,sha256=qIhpVmTphnXVt-ewdSJF6GlIih0C0ewhtGtOs9IgW3U,14625
101
+ onnx_diagnostic/tasks/text_generation.py,sha256=4Iqy24VmFAqTwiUkDRWaf5Oy6842FCSiiI8WQxnQGEw,14807
102
102
  onnx_diagnostic/tasks/text_to_image.py,sha256=mOS3Ruosi3hzRMxXLDN7ZkAbi7NnQb7MWwQP_okGVHs,2962
103
103
  onnx_diagnostic/tasks/zero_shot_image_classification.py,sha256=jJCMWuOqGv5ahCfjrcqxuYCJFhTgHV5KUf2yyv2yxYA,4624
104
104
  onnx_diagnostic/tasks/data/__init__.py,sha256=uJoemrWgEjI6oA-tMX7r3__x-b3siPmkgqaY7bgIles,401
105
105
  onnx_diagnostic/tasks/data/dummies_imagetext2text_generation_gemma3.onnx,sha256=UbtvmWMqcZOKJ-I-HXWI1A6YR6QDaFS5u_yXm5C3ZBw,10299
106
106
  onnx_diagnostic/torch_export_patches/__init__.py,sha256=0SaZedwznm1hQUCvXZsGZORV5vby954wEExr5faepGg,720
107
107
  onnx_diagnostic/torch_export_patches/onnx_export_errors.py,sha256=XHYtU7w3vsaTMCuF5X1YtOKxgwL8eEuktXzVZpRz55o,43431
108
- onnx_diagnostic/torch_export_patches/onnx_export_serialization.py,sha256=0HdubI06EGpxOICqDWZoVmZkVO9gAaFADEmby197EyM,11935
108
+ onnx_diagnostic/torch_export_patches/onnx_export_serialization.py,sha256=KZgr8WbLvmZte1G1hKA0zIlio5ZHz9MKIwGPqNamB6E,12690
109
109
  onnx_diagnostic/torch_export_patches/patch_details.py,sha256=UHBo4QTLF3ZgQ4951yYHIQqxOeRYNaG7x56XFcRTtg4,11794
110
110
  onnx_diagnostic/torch_export_patches/patch_expressions.py,sha256=VOsv71FsR_UZtxz4-5_VKL2sHQhOkHy9RkPJME2h7UU,3271
111
111
  onnx_diagnostic/torch_export_patches/patch_inputs.py,sha256=-TgcyjVzxTb5Y-_ibssTeaA5PFz6FJrV6q84HMUAsJw,8075
112
112
  onnx_diagnostic/torch_export_patches/patch_module.py,sha256=1Mn3xdpK1jSdRs6z1C-mJGkfGmD2TNRwLNoPaOW_EFI,40061
113
113
  onnx_diagnostic/torch_export_patches/patch_module_helper.py,sha256=2U0AdyZuU0W54QTdE7tY7imVzMnpQ5091ADNtTCkT8Y,6967
114
- onnx_diagnostic/torch_export_patches/eval/__init__.py,sha256=YQoOGt9XQLWqnJ15NnT7ri_jDevfvpuQwEJo38E-VRU,25056
114
+ onnx_diagnostic/torch_export_patches/eval/__init__.py,sha256=15FNHm0Ztz9kuHeXiL0dIT_ABaNtKh_jMRbM8Kt0h7Y,25057
115
115
  onnx_diagnostic/torch_export_patches/eval/model_cases.py,sha256=9h4yo9vKiK-E6zaXyAsxXGM-lCjd88ONybA1F3YcTI4,27988
116
116
  onnx_diagnostic/torch_export_patches/patches/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
117
117
  onnx_diagnostic/torch_export_patches/patches/_patch_transformers_attention.py,sha256=5JOyT95BNwHIuxaSFJDSEGsoI-6IwbgNnFwg2q3UM-Q,7731
@@ -129,11 +129,11 @@ onnx_diagnostic/torch_export_patches/patches/_patch_transformers_qwen3.py,sha256
129
129
  onnx_diagnostic/torch_export_patches/patches/_patch_transformers_rotary_embedding.py,sha256=LAqoL5SWISekZO15G1HIcCkN1JlBxGqb9XbK_eLzalA,16949
130
130
  onnx_diagnostic/torch_export_patches/patches/_patch_transformers_sam_mask_decoder.py,sha256=-6TuBm3sLAFEGuW3vRfOTtE5uP6aINFfu7xMnl27Dws,5703
131
131
  onnx_diagnostic/torch_export_patches/patches/patch_helper.py,sha256=kK_CGW643iVXxa-m6pttDBS7HTyMQaPypza7iqIInn4,721
132
- onnx_diagnostic/torch_export_patches/patches/patch_torch.py,sha256=VCs3uZHcuzosCqn9sSEskEWHJym_RrDJM6-G6FcTC08,45117
132
+ onnx_diagnostic/torch_export_patches/patches/patch_torch.py,sha256=IpvyM5yhA3tioyDAwAEe3gZDvbRUwCI_CzKmm76zwyU,45297
133
133
  onnx_diagnostic/torch_export_patches/patches/patch_transformers.py,sha256=1W3iKVYx2QT2xJTKlz1UmtjySuwv-rfT5yVL9DjOfzI,3376
134
134
  onnx_diagnostic/torch_export_patches/serialization/__init__.py,sha256=BHLdRPtNAtNPAS-bPKEj3-foGSPvwAbZXrHzGGPDLEw,1876
135
135
  onnx_diagnostic/torch_export_patches/serialization/diffusers_impl.py,sha256=drq3EH_yjcSuIWYsVeUWm8Cx6YCZFU6bP_1PLtPfY5I,945
136
- onnx_diagnostic/torch_export_patches/serialization/transformers_impl.py,sha256=sIHFvUQoMK8ytXQYB-k7OL62z8A3f5uDaq-S5R5uN-M,10034
136
+ onnx_diagnostic/torch_export_patches/serialization/transformers_impl.py,sha256=kcjMbGfNqftt5X1cP-d153qvK_d5FRRuDyCuieHGieU,10941
137
137
  onnx_diagnostic/torch_models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
138
138
  onnx_diagnostic/torch_models/code_sample.py,sha256=rCDZY64pkn6uIbJJSBuC5TlU_-uleI73I9GlbXtJd54,12923
139
139
  onnx_diagnostic/torch_models/llms.py,sha256=soyg4yC87ptGoeulJhKqw5opGmuLvH1pn_ZDXZ4Jr8E,90
@@ -141,19 +141,19 @@ onnx_diagnostic/torch_models/validate.py,sha256=JsEMv7aeg9tGGriKZ_CJeqGDfNUmZNdG
141
141
  onnx_diagnostic/torch_models/hghub/__init__.py,sha256=vi1Q7YHdddj1soiBN42MSvJdFqe2_KUoWafHISjwOu8,58
142
142
  onnx_diagnostic/torch_models/hghub/hub_api.py,sha256=V3azxUqb7mkmHQ8m5DCgg1WUU2NYBK12USEUy_sfYIA,14709
143
143
  onnx_diagnostic/torch_models/hghub/hub_data.py,sha256=6jR8A83cGP4Xw1Wg-q1zzKFpqzoVrybqm0Fm3yurkrE,9030
144
- onnx_diagnostic/torch_models/hghub/hub_data_cached_configs.py,sha256=Dxa13rsnTQ8eH_BcQvbY2bp1AYFtzuFrJ-J_urrSmeQ,292694
144
+ onnx_diagnostic/torch_models/hghub/hub_data_cached_configs.py,sha256=Q4rZz81erNAXzSoezR5kR1Qe_l_Rtn7k4egOx_L8LQU,292725
145
145
  onnx_diagnostic/torch_models/hghub/model_inputs.py,sha256=XahJ-m6ajdXg6vFGUOfV5IvFwn-yjAsIOU37nISbBoo,17646
146
146
  onnx_diagnostic/torch_models/hghub/model_specific.py,sha256=j50Nu7wddJMoqmD4QzMbNdFDUUgUmSBKRzPDH55TlUQ,2498
147
147
  onnx_diagnostic/torch_models/untrained/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
148
- onnx_diagnostic/torch_models/untrained/llm_phi2.py,sha256=y_akbdApi136qHcEQgykwIAYVw0Yfi0lbjb3DNuafaU,3948
148
+ onnx_diagnostic/torch_models/untrained/llm_phi2.py,sha256=HKj0krVNDI--mWHaTzagZ2RWAwe68P2n4JHKuW2ybFI,3975
149
149
  onnx_diagnostic/torch_models/untrained/llm_tiny_llm.py,sha256=QXw_Bs2SzfeiQMf-tmtVl83SmVOL4-Um7Qy-f0E48QI,2507
150
150
  onnx_diagnostic/torch_onnx/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
151
151
  onnx_diagnostic/torch_onnx/compare.py,sha256=O0lws4kzn8WAXr8-x-YMPr7oyBC9DtSIs4OfOr4S5-E,12305
152
152
  onnx_diagnostic/torch_onnx/runtime_info.py,sha256=u1bD6VXqzBCRmqmbzQtDswaPs1PH_ygr1r-CrcfXpNU,8562
153
153
  onnx_diagnostic/torch_onnx/sbs.py,sha256=8okBEIupMgw7TtKc80YFimMtwnY3GchdY05FsA9ooa0,40749
154
154
  onnx_diagnostic/torch_onnx/sbs_dataclasses.py,sha256=UctdBjzoPTQG1LS0tZ8A6E9hpoq5HWUYaJLPOPJc9FI,20299
155
- onnx_diagnostic-0.8.8.dist-info/licenses/LICENSE.txt,sha256=Vv6TXglX6Rc0d-f8aREhayhT-6PMQXEyOmI2NKlUCMc,1045
156
- onnx_diagnostic-0.8.8.dist-info/METADATA,sha256=sxPBIjOSLqQITi527dnl_3gk3RxT9kNCu47gPPyrg7Q,6905
157
- onnx_diagnostic-0.8.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
158
- onnx_diagnostic-0.8.8.dist-info/top_level.txt,sha256=KwNkXewmcobM3ZT1DJLVWH6ebJzA5qKg7cWqKfpGNT4,16
159
- onnx_diagnostic-0.8.8.dist-info/RECORD,,
155
+ onnx_diagnostic-0.8.10.dist-info/licenses/LICENSE.txt,sha256=Vv6TXglX6Rc0d-f8aREhayhT-6PMQXEyOmI2NKlUCMc,1045
156
+ onnx_diagnostic-0.8.10.dist-info/METADATA,sha256=dvG6WDkXq8Y56whBI15kB40qx1XhXSQyyUvN7Q6JOYo,6904
157
+ onnx_diagnostic-0.8.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
158
+ onnx_diagnostic-0.8.10.dist-info/top_level.txt,sha256=KwNkXewmcobM3ZT1DJLVWH6ebJzA5qKg7cWqKfpGNT4,16
159
+ onnx_diagnostic-0.8.10.dist-info/RECORD,,