onnx-diagnostic 0.8.8__py3-none-any.whl → 0.8.9__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.
- onnx_diagnostic/__init__.py +1 -1
- onnx_diagnostic/doc.py +258 -8
- onnx_diagnostic/export/api.py +478 -17
- onnx_diagnostic/export/dynamic_shapes.py +21 -6
- onnx_diagnostic/export/shape_helper.py +0 -8
- onnx_diagnostic/helpers/cache_helper.py +98 -13
- onnx_diagnostic/helpers/helper.py +6 -5
- onnx_diagnostic/helpers/onnx_helper.py +7 -0
- onnx_diagnostic/helpers/rt_helper.py +14 -1
- onnx_diagnostic/helpers/torch_helper.py +22 -9
- onnx_diagnostic/tasks/image_text_to_text.py +4 -1
- onnx_diagnostic/tasks/text_generation.py +17 -17
- onnx_diagnostic/torch_export_patches/eval/__init__.py +1 -1
- onnx_diagnostic/torch_export_patches/onnx_export_serialization.py +62 -38
- onnx_diagnostic/torch_export_patches/patches/patch_torch.py +12 -9
- onnx_diagnostic/torch_export_patches/serialization/transformers_impl.py +42 -30
- {onnx_diagnostic-0.8.8.dist-info → onnx_diagnostic-0.8.9.dist-info}/METADATA +2 -2
- {onnx_diagnostic-0.8.8.dist-info → onnx_diagnostic-0.8.9.dist-info}/RECORD +21 -21
- {onnx_diagnostic-0.8.8.dist-info → onnx_diagnostic-0.8.9.dist-info}/WHEEL +0 -0
- {onnx_diagnostic-0.8.8.dist-info → onnx_diagnostic-0.8.9.dist-info}/licenses/LICENSE.txt +0 -0
- {onnx_diagnostic-0.8.8.dist-info → onnx_diagnostic-0.8.9.dist-info}/top_level.txt +0 -0
|
@@ -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
|
-
|
|
105
|
-
) -> Tuple[List[Any], torch.utils._pytree.Context]:
|
|
106
|
-
|
|
107
|
-
|
|
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
|
-
|
|
112
|
-
) -> Tuple[List[Tuple[torch.utils._pytree.KeyEntry, Any]], torch.utils._pytree.Context]:
|
|
113
|
-
|
|
114
|
-
|
|
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
|
-
|
|
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
|
)
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: onnx-diagnostic
|
|
3
|
-
Version: 0.8.
|
|
3
|
+
Version: 0.8.9
|
|
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
|
|
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=
|
|
1
|
+
onnx_diagnostic/__init__.py,sha256=9vcutQbxSFR9MMn3ikZxIH3SQ1tgKjRC1yhyumWJ0Go,173
|
|
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
|
|
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=
|
|
12
|
+
onnx_diagnostic/export/api.py,sha256=iowYszK9ygdciW6P-C0isvW8ZAJW4yvI-3bjiWr2WAE,42945
|
|
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=
|
|
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=
|
|
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=
|
|
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=
|
|
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=
|
|
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=
|
|
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=
|
|
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=
|
|
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=
|
|
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=
|
|
108
|
+
onnx_diagnostic/torch_export_patches/onnx_export_serialization.py,sha256=zvvtKlqKiC0625vrKP3JNrUHCM3sFV_rPoKWH8Yq9OM,12523
|
|
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=
|
|
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=
|
|
132
|
+
onnx_diagnostic/torch_export_patches/patches/patch_torch.py,sha256=hia-wQRv2LNvUlb-GqML5JGu4a1LEs-lAvooQZefmyM,45196
|
|
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=
|
|
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
|
|
@@ -152,8 +152,8 @@ onnx_diagnostic/torch_onnx/compare.py,sha256=O0lws4kzn8WAXr8-x-YMPr7oyBC9DtSIs4O
|
|
|
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.
|
|
156
|
-
onnx_diagnostic-0.8.
|
|
157
|
-
onnx_diagnostic-0.8.
|
|
158
|
-
onnx_diagnostic-0.8.
|
|
159
|
-
onnx_diagnostic-0.8.
|
|
155
|
+
onnx_diagnostic-0.8.9.dist-info/licenses/LICENSE.txt,sha256=Vv6TXglX6Rc0d-f8aREhayhT-6PMQXEyOmI2NKlUCMc,1045
|
|
156
|
+
onnx_diagnostic-0.8.9.dist-info/METADATA,sha256=cVfclVmpVRwo-9HRCRdVH-YIvWjMWaJloPzmCyN_WOg,6903
|
|
157
|
+
onnx_diagnostic-0.8.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
|
158
|
+
onnx_diagnostic-0.8.9.dist-info/top_level.txt,sha256=KwNkXewmcobM3ZT1DJLVWH6ebJzA5qKg7cWqKfpGNT4,16
|
|
159
|
+
onnx_diagnostic-0.8.9.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|