onnx-diagnostic 0.7.1__py3-none-any.whl → 0.7.3__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/_command_lines_parser.py +22 -5
- onnx_diagnostic/ext_test_case.py +31 -0
- onnx_diagnostic/helpers/cache_helper.py +23 -12
- onnx_diagnostic/helpers/config_helper.py +16 -1
- onnx_diagnostic/helpers/log_helper.py +308 -83
- onnx_diagnostic/helpers/rt_helper.py +11 -1
- onnx_diagnostic/helpers/torch_helper.py +7 -3
- onnx_diagnostic/tasks/__init__.py +2 -0
- onnx_diagnostic/tasks/text_generation.py +17 -8
- onnx_diagnostic/tasks/text_to_image.py +91 -0
- onnx_diagnostic/torch_export_patches/eval/__init__.py +3 -1
- onnx_diagnostic/torch_export_patches/onnx_export_errors.py +24 -7
- onnx_diagnostic/torch_export_patches/onnx_export_serialization.py +148 -351
- onnx_diagnostic/torch_export_patches/patches/patch_transformers.py +89 -10
- onnx_diagnostic/torch_export_patches/serialization/__init__.py +46 -0
- onnx_diagnostic/torch_export_patches/serialization/diffusers_impl.py +34 -0
- onnx_diagnostic/torch_export_patches/serialization/transformers_impl.py +259 -0
- onnx_diagnostic/torch_models/hghub/hub_api.py +15 -4
- onnx_diagnostic/torch_models/hghub/hub_data.py +1 -0
- onnx_diagnostic/torch_models/hghub/hub_data_cached_configs.py +28 -0
- onnx_diagnostic/torch_models/hghub/model_inputs.py +24 -5
- onnx_diagnostic/torch_models/validate.py +36 -12
- {onnx_diagnostic-0.7.1.dist-info → onnx_diagnostic-0.7.3.dist-info}/METADATA +26 -1
- {onnx_diagnostic-0.7.1.dist-info → onnx_diagnostic-0.7.3.dist-info}/RECORD +28 -24
- {onnx_diagnostic-0.7.1.dist-info → onnx_diagnostic-0.7.3.dist-info}/WHEEL +0 -0
- {onnx_diagnostic-0.7.1.dist-info → onnx_diagnostic-0.7.3.dist-info}/licenses/LICENSE.txt +0 -0
- {onnx_diagnostic-0.7.1.dist-info → onnx_diagnostic-0.7.3.dist-info}/top_level.txt +0 -0
|
@@ -263,7 +263,7 @@ def validate_model(
|
|
|
263
263
|
use_pretrained: bool = False,
|
|
264
264
|
optimization: Optional[str] = None,
|
|
265
265
|
quiet: bool = False,
|
|
266
|
-
patch: bool = False,
|
|
266
|
+
patch: Union[bool, str, Dict[str, bool]] = False,
|
|
267
267
|
rewrite: bool = False,
|
|
268
268
|
stop_if_static: int = 1,
|
|
269
269
|
dump_folder: Optional[str] = None,
|
|
@@ -301,8 +301,10 @@ def validate_model(
|
|
|
301
301
|
:param optimization: optimization to apply to the exported model,
|
|
302
302
|
depend on the the exporter
|
|
303
303
|
:param quiet: if quiet, catches exception if any issue
|
|
304
|
-
:param patch: applies patches (``patch_transformers=True``)
|
|
305
|
-
|
|
304
|
+
:param patch: applies patches (``patch_transformers=True, path_diffusers=True``)
|
|
305
|
+
if True before exporting
|
|
306
|
+
see :func:`onnx_diagnostic.torch_export_patches.torch_export_patches`,
|
|
307
|
+
a string can be used to specify only one of them
|
|
306
308
|
:param rewrite: applies known rewriting (``patch_transformers=True``) before exporting,
|
|
307
309
|
see :func:`onnx_diagnostic.torch_export_patches.torch_export_patches`
|
|
308
310
|
:param stop_if_static: stops if a dynamic dimension becomes static,
|
|
@@ -346,9 +348,26 @@ def validate_model(
|
|
|
346
348
|
exported model returns the same outputs as the original one, otherwise,
|
|
347
349
|
:class:`onnx_diagnostic.reference.TorchOnnxEvaluator` is used.
|
|
348
350
|
"""
|
|
349
|
-
|
|
350
|
-
|
|
351
|
-
|
|
351
|
+
if isinstance(patch, bool):
|
|
352
|
+
patch_kwargs = (
|
|
353
|
+
dict(patch_transformers=True, patch_diffusers=True, patch=True)
|
|
354
|
+
if patch
|
|
355
|
+
else dict(patch=False)
|
|
356
|
+
)
|
|
357
|
+
elif isinstance(patch, str):
|
|
358
|
+
patch_kwargs = {"patch": True, **{p: True for p in patch.split(",")}} # noqa: C420
|
|
359
|
+
else:
|
|
360
|
+
assert isinstance(patch, dict), f"Unable to interpret patch={patch!r}"
|
|
361
|
+
patch_kwargs = patch.copy()
|
|
362
|
+
if "patch" not in patch_kwargs:
|
|
363
|
+
if any(patch_kwargs.values()):
|
|
364
|
+
patch_kwargs["patch"] = True
|
|
365
|
+
|
|
366
|
+
assert not rewrite or patch_kwargs.get("patch", False), (
|
|
367
|
+
f"rewrite={rewrite}, patch={patch}, patch_kwargs={patch_kwargs} "
|
|
368
|
+
f"patch must be True to enable rewriting, "
|
|
369
|
+
f"if --no-patch was specified on the command line, --no-rewrite must be added."
|
|
370
|
+
)
|
|
352
371
|
summary = version_summary()
|
|
353
372
|
summary.update(
|
|
354
373
|
dict(
|
|
@@ -361,6 +380,7 @@ def validate_model(
|
|
|
361
380
|
version_optimization=optimization or "",
|
|
362
381
|
version_quiet=str(quiet),
|
|
363
382
|
version_patch=str(patch),
|
|
383
|
+
version_patch_kwargs=str(patch_kwargs).replace(" ", ""),
|
|
364
384
|
version_rewrite=str(rewrite),
|
|
365
385
|
version_dump_folder=dump_folder or "",
|
|
366
386
|
version_drop_inputs=str(list(drop_inputs or "")),
|
|
@@ -396,7 +416,7 @@ def validate_model(
|
|
|
396
416
|
print(f"[validate_model] model_options={model_options!r}")
|
|
397
417
|
print(f"[validate_model] get dummy inputs with input_options={input_options}...")
|
|
398
418
|
print(
|
|
399
|
-
f"[validate_model] rewrite={rewrite},
|
|
419
|
+
f"[validate_model] rewrite={rewrite}, patch_kwargs={patch_kwargs}, "
|
|
400
420
|
f"stop_if_static={stop_if_static}"
|
|
401
421
|
)
|
|
402
422
|
print(f"[validate_model] exporter={exporter!r}, optimization={optimization!r}")
|
|
@@ -538,9 +558,13 @@ def validate_model(
|
|
|
538
558
|
if summary["model_module"] in sys.modules:
|
|
539
559
|
summary["model_file"] = str(sys.modules[summary["model_module"]].__file__) # type: ignore[index]
|
|
540
560
|
summary["model_config_class"] = data["configuration"].__class__.__name__
|
|
541
|
-
summary["model_config"] = str(
|
|
542
|
-
|
|
543
|
-
|
|
561
|
+
summary["model_config"] = str(
|
|
562
|
+
shrink_config(
|
|
563
|
+
data["configuration"]
|
|
564
|
+
if type(data["configuration"]) is dict
|
|
565
|
+
else data["configuration"].to_dict()
|
|
566
|
+
)
|
|
567
|
+
).replace(" ", "")
|
|
544
568
|
summary["model_id"] = model_id
|
|
545
569
|
|
|
546
570
|
if verbose:
|
|
@@ -568,18 +592,18 @@ def validate_model(
|
|
|
568
592
|
f"[validate_model] -- export the model with {exporter!r}, "
|
|
569
593
|
f"optimization={optimization!r}"
|
|
570
594
|
)
|
|
571
|
-
if
|
|
595
|
+
if patch_kwargs:
|
|
572
596
|
if verbose:
|
|
573
597
|
print(
|
|
574
598
|
f"[validate_model] applies patches before exporting "
|
|
575
599
|
f"stop_if_static={stop_if_static}"
|
|
576
600
|
)
|
|
577
601
|
with torch_export_patches( # type: ignore
|
|
578
|
-
patch_transformers=True,
|
|
579
602
|
stop_if_static=stop_if_static,
|
|
580
603
|
verbose=max(0, verbose - 1),
|
|
581
604
|
rewrite=data.get("rewrite", None),
|
|
582
605
|
dump_rewriting=(os.path.join(dump_folder, "rewrite") if dump_folder else None),
|
|
606
|
+
**patch_kwargs, # type: ignore[arg-type]
|
|
583
607
|
) as modificator:
|
|
584
608
|
data["inputs_export"] = modificator(data["inputs"]) # type: ignore
|
|
585
609
|
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: onnx-diagnostic
|
|
3
|
-
Version: 0.7.
|
|
3
|
+
Version: 0.7.3
|
|
4
4
|
Summary: Investigate ONNX models
|
|
5
5
|
Home-page: https://github.com/sdpython/onnx-diagnostic
|
|
6
6
|
Author: Xavier Dupré
|
|
@@ -64,13 +64,26 @@ onnx-diagnostic: investigate onnx models
|
|
|
64
64
|
|
|
65
65
|
The main feature is about `patches <https://github.com/sdpython/onnx-diagnostic/tree/main/onnx_diagnostic/torch_export_patches>`_:
|
|
66
66
|
it helps exporting **pytorch models into ONNX**, mostly designed for LLMs using dynamic caches.
|
|
67
|
+
Patches can be enabled as follows:
|
|
67
68
|
|
|
68
69
|
.. code-block:: python
|
|
69
70
|
|
|
71
|
+
from onnx_diagnostic.torch_export_patches import torch_export_patches
|
|
72
|
+
|
|
70
73
|
with torch_export_patches(patch_transformers=True) as f:
|
|
71
74
|
ep = torch.export.export(model, args, kwargs=kwargs, dynamic_shapes=dynamic_shapes)
|
|
72
75
|
# ...
|
|
73
76
|
|
|
77
|
+
Dynamic shapes are difficult to guess for caches, one function
|
|
78
|
+
returns a structure defining all dimensions as dynamic.
|
|
79
|
+
You need then to remove those which are not dynamic in your model.
|
|
80
|
+
|
|
81
|
+
.. code-block:: python
|
|
82
|
+
|
|
83
|
+
from onnx_diagnostic.export.shape_helper import all_dynamic_shape_from_inputs
|
|
84
|
+
|
|
85
|
+
dynamic_shapes = all_dynamic_shape_from_inputs(cache)
|
|
86
|
+
|
|
74
87
|
It also implements tools to investigate, validate exported models (ExportedProgramm, ONNXProgram, ...).
|
|
75
88
|
See `documentation of onnx-diagnostic <https://sdpython.github.io/doc/onnx-diagnostic/dev/>`_ and
|
|
76
89
|
`torch_export_patches <https://sdpython.github.io/doc/onnx-diagnostic/dev/api/torch_export_patches/index.html#onnx_diagnostic.torch_export_patches.torch_export_patches>`_.
|
|
@@ -127,14 +140,26 @@ Snapshot of usefuls tools
|
|
|
127
140
|
|
|
128
141
|
.. code-block:: python
|
|
129
142
|
|
|
143
|
+
from onnx_diagnostic.torch_export_patches import torch_export_patches
|
|
144
|
+
|
|
130
145
|
with torch_export_patches(patch_transformers=True) as f:
|
|
131
146
|
ep = torch.export.export(model, args, kwargs=kwargs, dynamic_shapes=dynamic_shapes)
|
|
132
147
|
# ...
|
|
133
148
|
|
|
149
|
+
**all_dynamic_shape_from_inputs**
|
|
150
|
+
|
|
151
|
+
.. code-block:: python
|
|
152
|
+
|
|
153
|
+
from onnx_diagnostic.export.shape_helper import all_dynamic_shape_from_inputs
|
|
154
|
+
|
|
155
|
+
dynamic_shapes = all_dynamic_shape_from_inputs(cache)
|
|
156
|
+
|
|
134
157
|
**torch_export_rewrite**
|
|
135
158
|
|
|
136
159
|
.. code-block:: python
|
|
137
160
|
|
|
161
|
+
from onnx_diagnostic.torch_export_patches import torch_export_rewrite
|
|
162
|
+
|
|
138
163
|
with torch_export_rewrite(rewrite=[Model.forward]) as f:
|
|
139
164
|
ep = torch.export.export(model, args, kwargs=kwargs, dynamic_shapes=dynamic_shapes)
|
|
140
165
|
# ...
|
|
@@ -1,9 +1,9 @@
|
|
|
1
|
-
onnx_diagnostic/__init__.py,sha256=
|
|
1
|
+
onnx_diagnostic/__init__.py,sha256=N1lf8_afRytDUnulPCeDVDPA-M4k7y9x7LbWwX0USZs,173
|
|
2
2
|
onnx_diagnostic/__main__.py,sha256=YmyV_Aq_ianDlHyKLHMa6h8YK3ZmFPpLVHLKjM91aCk,79
|
|
3
|
-
onnx_diagnostic/_command_lines_parser.py,sha256=
|
|
3
|
+
onnx_diagnostic/_command_lines_parser.py,sha256=TliHXedXFerv-zO6cBigxKbuKHE0-6TUhhsk1pdkz9M,28072
|
|
4
4
|
onnx_diagnostic/api.py,sha256=BhCl_yCd78N7TlVtPOHjeYv1QBEy39TjZ647rcHqLh0,345
|
|
5
5
|
onnx_diagnostic/doc.py,sha256=t3RELgfooYnVMAi0JSpggWkQEgUsREz8NmRvn0TnLI8,2829
|
|
6
|
-
onnx_diagnostic/ext_test_case.py,sha256=
|
|
6
|
+
onnx_diagnostic/ext_test_case.py,sha256=Bq4vdlM0P72H1orlKJTeOBqm1YGHTK-ylAlNsBe4LeA,43438
|
|
7
7
|
onnx_diagnostic/export/__init__.py,sha256=yEIoWiOeTwBsDhyYt2fTKuhtA0Ya1J9u9ZzMTOTWaWs,101
|
|
8
8
|
onnx_diagnostic/export/dynamic_shapes.py,sha256=HYf2OEi7PmRSj8uxMD-wbdVxxejkWXTPBAkxoFeM27A,40811
|
|
9
9
|
onnx_diagnostic/export/shape_helper.py,sha256=C9cEq_x8I40RKuD89qWIholN1XZoWhaKPfbZQhiPD3g,4725
|
|
@@ -11,19 +11,19 @@ onnx_diagnostic/export/validate.py,sha256=_PGUql2DJhIgGKo0WjTGUc5AgsZUx8fEs00MeP
|
|
|
11
11
|
onnx_diagnostic/helpers/__init__.py,sha256=GJ2GT7cgnlIveVUwMZhuvUwidbTJaKv8CsSIOpZDsJg,83
|
|
12
12
|
onnx_diagnostic/helpers/args_helper.py,sha256=SRWnqC7EENg09RZlA50B_PcdiIhdbgA4C3ACfzl5nMs,4419
|
|
13
13
|
onnx_diagnostic/helpers/bench_run.py,sha256=CGA6VMJZMH2gDhVueT9ypNm4PMcjGrrGFYp08nhWj9k,16539
|
|
14
|
-
onnx_diagnostic/helpers/cache_helper.py,sha256=
|
|
15
|
-
onnx_diagnostic/helpers/config_helper.py,sha256=
|
|
14
|
+
onnx_diagnostic/helpers/cache_helper.py,sha256=1slql7FdL89wzRWGoehLI7G1BY0cOsqGQmaZvLBvCIA,11229
|
|
15
|
+
onnx_diagnostic/helpers/config_helper.py,sha256=9h1NWC9RLmu43Yf5Cz9usjMdLiyLWXMhwgE4Lg-eOU8,3889
|
|
16
16
|
onnx_diagnostic/helpers/doc_helper.py,sha256=pl5MZd3_FaE8BqQnqoBuSBxoNCFcd2OJd3eITUSku5c,5897
|
|
17
17
|
onnx_diagnostic/helpers/graph_helper.py,sha256=hevQT5a7_QuriVPQcbT5qe18n99Doyl5h3-qshx1-uk,14093
|
|
18
18
|
onnx_diagnostic/helpers/helper.py,sha256=_6K0IvfK7ymBE8uWFAOA1ksU_fMvl2BRtlxj5SA9R2I,58203
|
|
19
|
-
onnx_diagnostic/helpers/log_helper.py,sha256=
|
|
19
|
+
onnx_diagnostic/helpers/log_helper.py,sha256=e89MI_i6PFBshm1cOOX5yCowEPIKzneMyCrpc34vpU0,77613
|
|
20
20
|
onnx_diagnostic/helpers/memory_peak.py,sha256=OT6mz0muBbBZY0pjgW2_eCk_lOtFRo-5w4jFo2Z6Kok,6380
|
|
21
21
|
onnx_diagnostic/helpers/mini_onnx_builder.py,sha256=p0Xh2Br38xAqUjB2214GiNOIbCgiVZKeyVEnjdyqyFI,21091
|
|
22
22
|
onnx_diagnostic/helpers/model_builder_helper.py,sha256=RvDyPFqRboEU3HsQV_xi9oy-o3_4KuGFVzs5MhksduY,12552
|
|
23
23
|
onnx_diagnostic/helpers/onnx_helper.py,sha256=pXXQjfyNTSUF-Kt72U4fnBDkYAnWYMxdSw8m0qk3xmE,39670
|
|
24
24
|
onnx_diagnostic/helpers/ort_session.py,sha256=UgUUeUslDxEFBc6w6f3HMq_a7bn4TBlItmojqWquSj4,29281
|
|
25
|
-
onnx_diagnostic/helpers/rt_helper.py,sha256=
|
|
26
|
-
onnx_diagnostic/helpers/torch_helper.py,sha256=
|
|
25
|
+
onnx_diagnostic/helpers/rt_helper.py,sha256=qbV6zyMs-iH6H65WHC2tu4h0psnHg0TX5fwfO_k-glg,4623
|
|
26
|
+
onnx_diagnostic/helpers/torch_helper.py,sha256=QfUXUPx0lZEqJBgyA97daPRDlT9duTM5Jq5Yjq1jJd8,32358
|
|
27
27
|
onnx_diagnostic/reference/__init__.py,sha256=rLZsxOlnb7-81F2CzepGnZLejaROg4JvgFaGR9FwVQA,208
|
|
28
28
|
onnx_diagnostic/reference/evaluator.py,sha256=RzNzjFDeMe-4X51Tb22N6aagazY5ktNq-mRmPcfY5EU,8848
|
|
29
29
|
onnx_diagnostic/reference/ort_evaluator.py,sha256=1O7dHj8Aspolidg6rB2Nm7hT3HaGb4TxAgjCCD0XVcQ,26159
|
|
@@ -71,7 +71,7 @@ onnx_diagnostic/reference/torch_ops/reduce_ops.py,sha256=9gFfraPTQbe_ZEUNCUis1JS
|
|
|
71
71
|
onnx_diagnostic/reference/torch_ops/sequence_ops.py,sha256=3EiVKpGfN4d1Iry4hgnr3MIJyEEKUrAIDgmRGsUXXa0,2297
|
|
72
72
|
onnx_diagnostic/reference/torch_ops/shape_ops.py,sha256=pJrNR2UB4PlWl6cv4EDl1uGl8YTBUUMQkhJcsh5K4sA,4291
|
|
73
73
|
onnx_diagnostic/reference/torch_ops/unary_ops.py,sha256=E8Ys1eZsOTsucBKoXb1_Kl5LbBDygniDvW2BvN4IPMo,1708
|
|
74
|
-
onnx_diagnostic/tasks/__init__.py,sha256=
|
|
74
|
+
onnx_diagnostic/tasks/__init__.py,sha256=0BYtrAnr0zKN3om71oi-OVz5wFYDp9WWIk51qWjjyCw,2450
|
|
75
75
|
onnx_diagnostic/tasks/automatic_speech_recognition.py,sha256=7OspFypNHLSL6huvP9ms_euhHqYXyTZjAJ8Z7EXGimk,6967
|
|
76
76
|
onnx_diagnostic/tasks/feature_extraction.py,sha256=CbxbGsv3JvEQ2J9tO2DOpMHcJj5ZlCwY81ZB3hPB4D4,2339
|
|
77
77
|
onnx_diagnostic/tasks/fill_mask.py,sha256=ZWz8swzEeRbkmbY9oZ4CM1LYCWWUxnS5CqrKmUVw-u0,2457
|
|
@@ -83,36 +83,40 @@ onnx_diagnostic/tasks/sentence_similarity.py,sha256=3MvNxjC1iEMtQL_jH1c8bmrVc5IG
|
|
|
83
83
|
onnx_diagnostic/tasks/summarization.py,sha256=NLwqhpiQrU8UWd3u30VsNA3FsL315S3nlQ7ycUzJueo,8105
|
|
84
84
|
onnx_diagnostic/tasks/text2text_generation.py,sha256=mYvsq-O69fr5pitX0mWugT76QuK4xUs40Vsz9ru_XK8,8522
|
|
85
85
|
onnx_diagnostic/tasks/text_classification.py,sha256=NCCKobBQyCc7dSVj7_5N6S_RuvBlRMAdWkS2rVvrzck,2528
|
|
86
|
-
onnx_diagnostic/tasks/text_generation.py,sha256=
|
|
86
|
+
onnx_diagnostic/tasks/text_generation.py,sha256=fraa5w2_eexXLAcDAR1e4g9zWKdvWybMYi1zu7XU7J4,12988
|
|
87
|
+
onnx_diagnostic/tasks/text_to_image.py,sha256=6z-rFG6MX9aBi8YoYtYI_8OV3M3Tfoi45V8S9csQ7k4,2799
|
|
87
88
|
onnx_diagnostic/tasks/zero_shot_image_classification.py,sha256=GKaXm8g7cK23h3wJEUc6Q-6mpmLAzQ4YkJbd-eGP7Y4,4496
|
|
88
89
|
onnx_diagnostic/torch_export_patches/__init__.py,sha256=0SaZedwznm1hQUCvXZsGZORV5vby954wEExr5faepGg,720
|
|
89
|
-
onnx_diagnostic/torch_export_patches/onnx_export_errors.py,sha256=
|
|
90
|
-
onnx_diagnostic/torch_export_patches/onnx_export_serialization.py,sha256=
|
|
90
|
+
onnx_diagnostic/torch_export_patches/onnx_export_errors.py,sha256=h_txSp30QmF1R_Q2wL4qpPqY59Dund2P9nAAsvucS8A,21245
|
|
91
|
+
onnx_diagnostic/torch_export_patches/onnx_export_serialization.py,sha256=wFE2fNxihAA3iua79AEB97_RBVv4wvGxwS9g4RJaSIc,10715
|
|
91
92
|
onnx_diagnostic/torch_export_patches/patch_expressions.py,sha256=vr4tt61cbDnaaaduzMj4UBZ8OUtr6GfDpIWwOYqjWzs,3213
|
|
92
93
|
onnx_diagnostic/torch_export_patches/patch_inputs.py,sha256=9b4pmyT00BwLqi7WG-gliep1RUy3gXEgW6BDnlSSA-M,7689
|
|
93
94
|
onnx_diagnostic/torch_export_patches/patch_module.py,sha256=R2d9IHM-RwsBKDsxuBIJnEqMoxbS9gd4YWFGG2wwV5A,39881
|
|
94
95
|
onnx_diagnostic/torch_export_patches/patch_module_helper.py,sha256=2U0AdyZuU0W54QTdE7tY7imVzMnpQ5091ADNtTCkT8Y,6967
|
|
95
|
-
onnx_diagnostic/torch_export_patches/eval/__init__.py,sha256=
|
|
96
|
+
onnx_diagnostic/torch_export_patches/eval/__init__.py,sha256=57x62uZNA80XiWgkG8Fe0_8YJcIVrvKLPqvwLDPJwgc,24008
|
|
96
97
|
onnx_diagnostic/torch_export_patches/eval/model_cases.py,sha256=DTvdHPtNQh25Akv5o3D4Jxf1L1-SJ7w14tgvj8AAns8,26577
|
|
97
98
|
onnx_diagnostic/torch_export_patches/patches/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
98
99
|
onnx_diagnostic/torch_export_patches/patches/patch_torch.py,sha256=KaZ8TjDa9ATgT4HllYzzoNf_51q_yOj_GuF5NYjPCrU,18913
|
|
99
|
-
onnx_diagnostic/torch_export_patches/patches/patch_transformers.py,sha256=
|
|
100
|
+
onnx_diagnostic/torch_export_patches/patches/patch_transformers.py,sha256=QGfXf1IlsOToQKF9NOHau_l62u1alQbJ6KeiSKFb980,44061
|
|
101
|
+
onnx_diagnostic/torch_export_patches/serialization/__init__.py,sha256=BHLdRPtNAtNPAS-bPKEj3-foGSPvwAbZXrHzGGPDLEw,1876
|
|
102
|
+
onnx_diagnostic/torch_export_patches/serialization/diffusers_impl.py,sha256=drq3EH_yjcSuIWYsVeUWm8Cx6YCZFU6bP_1PLtPfY5I,945
|
|
103
|
+
onnx_diagnostic/torch_export_patches/serialization/transformers_impl.py,sha256=9u2jkqnuyBkIF3R2sDEO0Jlkedl-cQhBNXxXXDLSEwE,8885
|
|
100
104
|
onnx_diagnostic/torch_models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
101
105
|
onnx_diagnostic/torch_models/llms.py,sha256=soyg4yC87ptGoeulJhKqw5opGmuLvH1pn_ZDXZ4Jr8E,90
|
|
102
|
-
onnx_diagnostic/torch_models/validate.py,sha256=
|
|
106
|
+
onnx_diagnostic/torch_models/validate.py,sha256=uUWkrCNE766c6BYKe-g5jQW3w89cvb7F-xnIkKsZKSw,63254
|
|
103
107
|
onnx_diagnostic/torch_models/hghub/__init__.py,sha256=vi1Q7YHdddj1soiBN42MSvJdFqe2_KUoWafHISjwOu8,58
|
|
104
|
-
onnx_diagnostic/torch_models/hghub/hub_api.py,sha256=
|
|
105
|
-
onnx_diagnostic/torch_models/hghub/hub_data.py,sha256=
|
|
106
|
-
onnx_diagnostic/torch_models/hghub/hub_data_cached_configs.py,sha256=
|
|
107
|
-
onnx_diagnostic/torch_models/hghub/model_inputs.py,sha256=
|
|
108
|
+
onnx_diagnostic/torch_models/hghub/hub_api.py,sha256=Bvr-sTAhS6s6UCkt-KsY_7Mdai08-AQzvHrzbYCSuvk,13186
|
|
109
|
+
onnx_diagnostic/torch_models/hghub/hub_data.py,sha256=NTTDsCtIVvYnr5J3rlcq0GSGDOzOPzq9Tsnb3oVf4q8,8309
|
|
110
|
+
onnx_diagnostic/torch_models/hghub/hub_data_cached_configs.py,sha256=zZvIxTbmL55x44kCj3-T5Kg3Qzm9KB_Xj-MCcU9-LuQ,268245
|
|
111
|
+
onnx_diagnostic/torch_models/hghub/model_inputs.py,sha256=tuTGfcFKK3ugA9Fh909THHWkrNyXkU5IhV6RumlNUiQ,10830
|
|
108
112
|
onnx_diagnostic/torch_models/untrained/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
109
113
|
onnx_diagnostic/torch_models/untrained/llm_phi2.py,sha256=ynBTDHJHCk44NjLT_t6OiFDBdPP0rFGPteiONDxvztw,3708
|
|
110
114
|
onnx_diagnostic/torch_models/untrained/llm_tiny_llm.py,sha256=QXw_Bs2SzfeiQMf-tmtVl83SmVOL4-Um7Qy-f0E48QI,2507
|
|
111
115
|
onnx_diagnostic/torch_onnx/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
112
116
|
onnx_diagnostic/torch_onnx/runtime_info.py,sha256=1g9F_Jf9AAgYQU4stbsrFXwQl-30mWlQrFbQ7val8Ps,9268
|
|
113
117
|
onnx_diagnostic/torch_onnx/sbs.py,sha256=1EL25DeYFzlBSiFG_XjePBLvsiItRXbdDrr5-QZW2mA,16878
|
|
114
|
-
onnx_diagnostic-0.7.
|
|
115
|
-
onnx_diagnostic-0.7.
|
|
116
|
-
onnx_diagnostic-0.7.
|
|
117
|
-
onnx_diagnostic-0.7.
|
|
118
|
-
onnx_diagnostic-0.7.
|
|
118
|
+
onnx_diagnostic-0.7.3.dist-info/licenses/LICENSE.txt,sha256=Vv6TXglX6Rc0d-f8aREhayhT-6PMQXEyOmI2NKlUCMc,1045
|
|
119
|
+
onnx_diagnostic-0.7.3.dist-info/METADATA,sha256=uMYdGJbm6K04yoi5UidH3uA0nK_PXrnVXe83s9v6yPE,7431
|
|
120
|
+
onnx_diagnostic-0.7.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
|
121
|
+
onnx_diagnostic-0.7.3.dist-info/top_level.txt,sha256=KwNkXewmcobM3ZT1DJLVWH6ebJzA5qKg7cWqKfpGNT4,16
|
|
122
|
+
onnx_diagnostic-0.7.3.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|