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.
Files changed (28) hide show
  1. onnx_diagnostic/__init__.py +1 -1
  2. onnx_diagnostic/_command_lines_parser.py +22 -5
  3. onnx_diagnostic/ext_test_case.py +31 -0
  4. onnx_diagnostic/helpers/cache_helper.py +23 -12
  5. onnx_diagnostic/helpers/config_helper.py +16 -1
  6. onnx_diagnostic/helpers/log_helper.py +308 -83
  7. onnx_diagnostic/helpers/rt_helper.py +11 -1
  8. onnx_diagnostic/helpers/torch_helper.py +7 -3
  9. onnx_diagnostic/tasks/__init__.py +2 -0
  10. onnx_diagnostic/tasks/text_generation.py +17 -8
  11. onnx_diagnostic/tasks/text_to_image.py +91 -0
  12. onnx_diagnostic/torch_export_patches/eval/__init__.py +3 -1
  13. onnx_diagnostic/torch_export_patches/onnx_export_errors.py +24 -7
  14. onnx_diagnostic/torch_export_patches/onnx_export_serialization.py +148 -351
  15. onnx_diagnostic/torch_export_patches/patches/patch_transformers.py +89 -10
  16. onnx_diagnostic/torch_export_patches/serialization/__init__.py +46 -0
  17. onnx_diagnostic/torch_export_patches/serialization/diffusers_impl.py +34 -0
  18. onnx_diagnostic/torch_export_patches/serialization/transformers_impl.py +259 -0
  19. onnx_diagnostic/torch_models/hghub/hub_api.py +15 -4
  20. onnx_diagnostic/torch_models/hghub/hub_data.py +1 -0
  21. onnx_diagnostic/torch_models/hghub/hub_data_cached_configs.py +28 -0
  22. onnx_diagnostic/torch_models/hghub/model_inputs.py +24 -5
  23. onnx_diagnostic/torch_models/validate.py +36 -12
  24. {onnx_diagnostic-0.7.1.dist-info → onnx_diagnostic-0.7.3.dist-info}/METADATA +26 -1
  25. {onnx_diagnostic-0.7.1.dist-info → onnx_diagnostic-0.7.3.dist-info}/RECORD +28 -24
  26. {onnx_diagnostic-0.7.1.dist-info → onnx_diagnostic-0.7.3.dist-info}/WHEEL +0 -0
  27. {onnx_diagnostic-0.7.1.dist-info → onnx_diagnostic-0.7.3.dist-info}/licenses/LICENSE.txt +0 -0
  28. {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``) before exporting,
305
- see :func:`onnx_diagnostic.torch_export_patches.torch_export_patches`
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
- assert (
350
- not rewrite or patch
351
- ), f"rewrite={rewrite}, patch={patch}, patch must be True to enable rewriting"
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}, patch={patch}, "
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(shrink_config(data["configuration"].to_dict())).replace(
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 patch:
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.1
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=YwjIZRhfTzRgTOBvmUSNNYX0SBBdmLsWfkMVwHkJloQ,173
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=WpCri2dqc1a1KYthQcb4-eN0htfiWeLrAkncNK3cZaY,27466
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=IX-DNabvsPw8UkUeXC1amw3nnzdmJ3DeERn4E1Y_omo,42416
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=E_D0z5ks4zUJb9H6K19MKhUZ-nZTC_dgeDO5zXNFE9g,10824
15
- onnx_diagnostic/helpers/config_helper.py,sha256=CdMeUhmDe0LfKcdPv9-Lzt73RRs29NmUHg9uVrdFwTQ,3479
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=qZdvHHQqkYdZOf8UsIrByswMYSF_axca27JXRyQk52Y,69163
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=BXU_u1syk2RyM0HTFHKEiO6rHHhZW2UFPyUTVdeq8BU,4251
26
- onnx_diagnostic/helpers/torch_helper.py,sha256=L7qv14q4r1LcDKpEVobhySK6VE_X3h88Acvr6Kt4qEk,32244
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=5XXM-rv-Hk2gSHvqsww9DzVd9mcRifacgcPgvPCjnDM,2412
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=PDh870BB-llzlu8h_aZX4Z-9QLzcGmDwX5aKJPy_K90,12504
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=op8jgnTa_1T_bGN172A6YFTtkQv_ALMNu1oukrsFt9U,20634
90
- onnx_diagnostic/torch_export_patches/onnx_export_serialization.py,sha256=Nsf7HUJqu3ZRd0o9vUVCF6ifmS5UaQM6hB_Gmn19dNI,17095
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=VtkQB1o3Q2Fh99OOF6vQ2dynkhwzx2Wx6oB-rRbvTI0,23954
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=GwcPUaSm-Zys2pWHac8Wcvpmy2h4oiFQDmx_D3GZNBA,41007
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=NXFmJKmoO4reaeiu2ibuVgMRLS-l0WSdLhjn40_YsbU,62177
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=q4jUgSJ8AD28mpX7yDAUp0z7EQgb8euuD-P9Hayehds,12672
105
- onnx_diagnostic/torch_models/hghub/hub_data.py,sha256=lM9IDb5-3X8NSHcSPJFLS3tAvu_FqvcetyoHn-P2FIM,8272
106
- onnx_diagnostic/torch_models/hghub/hub_data_cached_configs.py,sha256=dE8tHksGOsTk77jpa7mldLYzaQ5joKxDxDB0ZnwQBV4,267246
107
- onnx_diagnostic/torch_models/hghub/model_inputs.py,sha256=XCllU8_fJBjkCGSV8cdqlpF1QH6AN_OAErK0aAXNQts,10261
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.1.dist-info/licenses/LICENSE.txt,sha256=Vv6TXglX6Rc0d-f8aREhayhT-6PMQXEyOmI2NKlUCMc,1045
115
- onnx_diagnostic-0.7.1.dist-info/METADATA,sha256=7YUf_2f3gFaGkWUbOrZNqREtnQSpRXx5q65HJEDcDI8,6631
116
- onnx_diagnostic-0.7.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
117
- onnx_diagnostic-0.7.1.dist-info/top_level.txt,sha256=KwNkXewmcobM3ZT1DJLVWH6ebJzA5qKg7cWqKfpGNT4,16
118
- onnx_diagnostic-0.7.1.dist-info/RECORD,,
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,,