microsoft/onnxruntime-extensions
Publicmirrored from https://github.com/microsoft/onnxruntime-extensionsAvailable
onnxruntime_extensions/pnp/__init__.py
13lines · modecode
| 1 | # onnxruntime-extensions pre&post processing frontend depends on the PyTorch |
| 2 | try: |
| 3 | import torch |
| 4 | except ImportError as e: |
| 5 | print("No torch installation found, which is required by the pre&post scripting!") |
| 6 | raise e |
| 7 | |
| 8 | from ._base import ProcessingTracedModule, ProcessingScriptModule, CustomFunction |
| 9 | from ._torchext import * # noqa |
| 10 | from ._unifier import export |
| 11 | |
| 12 | from ._imagenet import * # noqa |
| 13 | from ._nlp import PreHuggingFaceGPT2, PreHuggingFaceBert, HfBertTokenizer # noqa |