Datasetpublic

codekingpro/portable-devtools

sourceHugging Faceupdated 5mo agoView on Hugging Face

1likes14kdownloads

rootpythonuser_packagesPython313site-packagesonnxruntimetools__init__.py

__init__.py11 linesDownload Raw Back to tools

1# -------------------------------------------------------------------------
2# Copyright (c) Microsoft Corporation. All rights reserved.
3# Licensed under the MIT License.
4# --------------------------------------------------------------------------
5# appended to the __init__.py in the onnxruntime module's 'tools' folder from /tools/python/util/__init__append.py
6import importlib.util
7
8have_torch = importlib.util.find_spec("torch")
9if have_torch:
10    from .pytorch_export_helpers import infer_input_info  # noqa: F401
11