m2m模型翻译
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

10 lines
512 B

6 months ago
  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
  6. import importlib.util
  7. have_torch = importlib.util.find_spec("torch")
  8. if have_torch:
  9. from .pytorch_export_helpers import infer_input_info # noqa