├── .github ├── dependabot.yml └── workflows │ ├── build_documentation.yml │ ├── build_pr_documentation.yml │ ├── quality.yml │ ├── style_bot.yml │ ├── test_cli.yml │ ├── test_exporters_common.yml │ ├── test_exporters_onnx.yml │ ├── test_exporters_onnx_cli.yml │ ├── test_offline.yml │ ├── test_onnx.yml │ ├── test_onnxruntime.yml │ ├── test_onnxruntime_gpu.yml │ ├── test_onnxruntime_slow.yml │ └── upload_pr_documentation.yml ├── .gitignore ├── LICENSE ├── Makefile ├── README.md ├── docs └── source │ ├── _toctree.yml │ ├── index.mdx │ ├── installation.mdx │ ├── onnx │ ├── overview.mdx │ ├── package_reference │ │ ├── configuration.mdx │ │ └── export.mdx │ └── usage_guides │ │ ├── contribute.mdx │ │ └── export_a_model.mdx │ ├── onnxruntime │ ├── concept_guides │ │ └── onnx.mdx │ ├── overview.mdx │ ├── package_reference │ │ ├── configuration.mdx │ │ ├── modeling.mdx │ │ ├── modeling_diffusion.mdx │ │ ├── optimization.mdx │ │ ├── pipelines.mdx │ │ └── quantization.mdx │ ├── quickstart.mdx │ └── usage_guides │ │ ├── amdgpu.mdx │ │ ├── gpu.mdx │ │ ├── models.mdx │ │ ├── optimization.mdx │ │ ├── pipelines.mdx │ │ └── quantization.mdx │ └── quickstart.mdx ├── examples ├── gemma3.py └── onnxruntime │ ├── optimization │ ├── multiple-choice │ │ ├── README.md │ │ ├── requirements.txt │ │ └── run_swag.py │ ├── question-answering │ │ ├── README.md │ │ ├── requirements.txt │ │ ├── run_qa.py │ │ ├── trainer_qa.py │ │ └── utils_qa.py │ ├── text-classification │ │ ├── README.md │ │ ├── requirements.txt │ │ └── run_glue.py │ └── token-classification │ │ ├── README.md │ │ ├── requirements.txt │ │ └── run_ner.py │ └── quantization │ ├── image-classification │ ├── README.md │ ├── requirements.txt │ └── run_image_classification.py │ ├── multiple-choice │ ├── README.md │ ├── requirements.txt │ └── run_swag.py │ ├── question-answering │ ├── README.md │ ├── requirements.txt │ ├── run_qa.py │ ├── trainer_qa.py │ └── utils_qa.py │ ├── text-classification │ ├── README.md │ ├── requirements.txt │ └── run_glue.py │ └── token-classification │ ├── README.md │ ├── requirements.txt │ └── run_ner.py ├── optimum ├── commands │ ├── export │ │ └── onnx.py │ ├── onnxruntime │ │ ├── base.py │ │ ├── optimize.py │ │ └── quantize.py │ └── register │ │ ├── register_export.py │ │ └── register_onnxruntime.py ├── exporters │ └── onnx │ │ ├── __init__.py │ │ ├── __main__.py │ │ ├── _traceable_cache.py │ │ ├── _traceable_decorator.py │ │ ├── base.py │ │ ├── config.py │ │ ├── constants.py │ │ ├── convert.py │ │ ├── input_generators.py │ │ ├── model_configs.py │ │ ├── model_patcher.py │ │ └── utils.py ├── onnx │ ├── __init__.py │ ├── graph_transformations.py │ ├── transformations_utils.py │ ├── utils.py │ └── version.py └── onnxruntime │ ├── __init__.py │ ├── base.py │ ├── configuration.py │ ├── constants.py │ ├── dummy_objects.py │ ├── modeling.py │ ├── modeling_decoder.py │ ├── modeling_diffusion.py │ ├── modeling_ort.py │ ├── modeling_seq2seq.py │ ├── optimization.py │ ├── pipelines.py │ ├── preprocessors │ ├── __init__.py │ ├── passes │ │ ├── __init__.py │ │ ├── excluders.py │ │ ├── fully_connected.py │ │ ├── gelu.py │ │ └── layernorm.py │ └── quantization.py │ ├── quantization.py │ └── utils.py ├── pyproject.toml └── tests ├── assets └── onnx │ ├── config.json │ └── model.onnx ├── cli └── test_cli.py ├── exporters ├── common │ └── test_tasks_manager.py └── onnx │ ├── test_export.py │ ├── test_export_cli.py │ └── utils_tests.py ├── onnx ├── test_onnx_export_custom_module.py └── test_onnx_graph_transformations.py └── onnxruntime ├── test_decoder.py ├── test_diffusion.py ├── test_modeling.py ├── test_optimization.py ├── test_quantization.py ├── test_seq2seq.py ├── test_timm.py ├── test_utils.py └── testing_utils.py /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/workflows/build_documentation.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/.github/workflows/build_documentation.yml -------------------------------------------------------------------------------- /.github/workflows/build_pr_documentation.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/.github/workflows/build_pr_documentation.yml -------------------------------------------------------------------------------- /.github/workflows/quality.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/.github/workflows/quality.yml -------------------------------------------------------------------------------- /.github/workflows/style_bot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/.github/workflows/style_bot.yml -------------------------------------------------------------------------------- /.github/workflows/test_cli.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/.github/workflows/test_cli.yml -------------------------------------------------------------------------------- /.github/workflows/test_exporters_common.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/.github/workflows/test_exporters_common.yml -------------------------------------------------------------------------------- /.github/workflows/test_exporters_onnx.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/.github/workflows/test_exporters_onnx.yml -------------------------------------------------------------------------------- /.github/workflows/test_exporters_onnx_cli.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/.github/workflows/test_exporters_onnx_cli.yml -------------------------------------------------------------------------------- /.github/workflows/test_offline.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/.github/workflows/test_offline.yml -------------------------------------------------------------------------------- /.github/workflows/test_onnx.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/.github/workflows/test_onnx.yml -------------------------------------------------------------------------------- /.github/workflows/test_onnxruntime.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/.github/workflows/test_onnxruntime.yml -------------------------------------------------------------------------------- /.github/workflows/test_onnxruntime_gpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/.github/workflows/test_onnxruntime_gpu.yml -------------------------------------------------------------------------------- /.github/workflows/test_onnxruntime_slow.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/.github/workflows/test_onnxruntime_slow.yml -------------------------------------------------------------------------------- /.github/workflows/upload_pr_documentation.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/.github/workflows/upload_pr_documentation.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/README.md -------------------------------------------------------------------------------- /docs/source/_toctree.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/_toctree.yml -------------------------------------------------------------------------------- /docs/source/index.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/index.mdx -------------------------------------------------------------------------------- /docs/source/installation.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/installation.mdx -------------------------------------------------------------------------------- /docs/source/onnx/overview.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/onnx/overview.mdx -------------------------------------------------------------------------------- /docs/source/onnx/package_reference/configuration.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/onnx/package_reference/configuration.mdx -------------------------------------------------------------------------------- /docs/source/onnx/package_reference/export.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/onnx/package_reference/export.mdx -------------------------------------------------------------------------------- /docs/source/onnx/usage_guides/contribute.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/onnx/usage_guides/contribute.mdx -------------------------------------------------------------------------------- /docs/source/onnx/usage_guides/export_a_model.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/onnx/usage_guides/export_a_model.mdx -------------------------------------------------------------------------------- /docs/source/onnxruntime/concept_guides/onnx.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/onnxruntime/concept_guides/onnx.mdx -------------------------------------------------------------------------------- /docs/source/onnxruntime/overview.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/onnxruntime/overview.mdx -------------------------------------------------------------------------------- /docs/source/onnxruntime/package_reference/configuration.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/onnxruntime/package_reference/configuration.mdx -------------------------------------------------------------------------------- /docs/source/onnxruntime/package_reference/modeling.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/onnxruntime/package_reference/modeling.mdx -------------------------------------------------------------------------------- /docs/source/onnxruntime/package_reference/modeling_diffusion.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/onnxruntime/package_reference/modeling_diffusion.mdx -------------------------------------------------------------------------------- /docs/source/onnxruntime/package_reference/optimization.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/onnxruntime/package_reference/optimization.mdx -------------------------------------------------------------------------------- /docs/source/onnxruntime/package_reference/pipelines.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/onnxruntime/package_reference/pipelines.mdx -------------------------------------------------------------------------------- /docs/source/onnxruntime/package_reference/quantization.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/onnxruntime/package_reference/quantization.mdx -------------------------------------------------------------------------------- /docs/source/onnxruntime/quickstart.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/onnxruntime/quickstart.mdx -------------------------------------------------------------------------------- /docs/source/onnxruntime/usage_guides/amdgpu.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/onnxruntime/usage_guides/amdgpu.mdx -------------------------------------------------------------------------------- /docs/source/onnxruntime/usage_guides/gpu.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/onnxruntime/usage_guides/gpu.mdx -------------------------------------------------------------------------------- /docs/source/onnxruntime/usage_guides/models.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/onnxruntime/usage_guides/models.mdx -------------------------------------------------------------------------------- /docs/source/onnxruntime/usage_guides/optimization.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/onnxruntime/usage_guides/optimization.mdx -------------------------------------------------------------------------------- /docs/source/onnxruntime/usage_guides/pipelines.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/onnxruntime/usage_guides/pipelines.mdx -------------------------------------------------------------------------------- /docs/source/onnxruntime/usage_guides/quantization.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/onnxruntime/usage_guides/quantization.mdx -------------------------------------------------------------------------------- /docs/source/quickstart.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/docs/source/quickstart.mdx -------------------------------------------------------------------------------- /examples/gemma3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/gemma3.py -------------------------------------------------------------------------------- /examples/onnxruntime/optimization/multiple-choice/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/optimization/multiple-choice/README.md -------------------------------------------------------------------------------- /examples/onnxruntime/optimization/multiple-choice/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/optimization/multiple-choice/requirements.txt -------------------------------------------------------------------------------- /examples/onnxruntime/optimization/multiple-choice/run_swag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/optimization/multiple-choice/run_swag.py -------------------------------------------------------------------------------- /examples/onnxruntime/optimization/question-answering/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/optimization/question-answering/README.md -------------------------------------------------------------------------------- /examples/onnxruntime/optimization/question-answering/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/optimization/question-answering/requirements.txt -------------------------------------------------------------------------------- /examples/onnxruntime/optimization/question-answering/run_qa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/optimization/question-answering/run_qa.py -------------------------------------------------------------------------------- /examples/onnxruntime/optimization/question-answering/trainer_qa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/optimization/question-answering/trainer_qa.py -------------------------------------------------------------------------------- /examples/onnxruntime/optimization/question-answering/utils_qa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/optimization/question-answering/utils_qa.py -------------------------------------------------------------------------------- /examples/onnxruntime/optimization/text-classification/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/optimization/text-classification/README.md -------------------------------------------------------------------------------- /examples/onnxruntime/optimization/text-classification/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/optimization/text-classification/requirements.txt -------------------------------------------------------------------------------- /examples/onnxruntime/optimization/text-classification/run_glue.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/optimization/text-classification/run_glue.py -------------------------------------------------------------------------------- /examples/onnxruntime/optimization/token-classification/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/optimization/token-classification/README.md -------------------------------------------------------------------------------- /examples/onnxruntime/optimization/token-classification/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/optimization/token-classification/requirements.txt -------------------------------------------------------------------------------- /examples/onnxruntime/optimization/token-classification/run_ner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/optimization/token-classification/run_ner.py -------------------------------------------------------------------------------- /examples/onnxruntime/quantization/image-classification/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/quantization/image-classification/README.md -------------------------------------------------------------------------------- /examples/onnxruntime/quantization/image-classification/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/quantization/image-classification/requirements.txt -------------------------------------------------------------------------------- /examples/onnxruntime/quantization/image-classification/run_image_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/quantization/image-classification/run_image_classification.py -------------------------------------------------------------------------------- /examples/onnxruntime/quantization/multiple-choice/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/quantization/multiple-choice/README.md -------------------------------------------------------------------------------- /examples/onnxruntime/quantization/multiple-choice/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/quantization/multiple-choice/requirements.txt -------------------------------------------------------------------------------- /examples/onnxruntime/quantization/multiple-choice/run_swag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/quantization/multiple-choice/run_swag.py -------------------------------------------------------------------------------- /examples/onnxruntime/quantization/question-answering/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/quantization/question-answering/README.md -------------------------------------------------------------------------------- /examples/onnxruntime/quantization/question-answering/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/quantization/question-answering/requirements.txt -------------------------------------------------------------------------------- /examples/onnxruntime/quantization/question-answering/run_qa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/quantization/question-answering/run_qa.py -------------------------------------------------------------------------------- /examples/onnxruntime/quantization/question-answering/trainer_qa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/quantization/question-answering/trainer_qa.py -------------------------------------------------------------------------------- /examples/onnxruntime/quantization/question-answering/utils_qa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/quantization/question-answering/utils_qa.py -------------------------------------------------------------------------------- /examples/onnxruntime/quantization/text-classification/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/quantization/text-classification/README.md -------------------------------------------------------------------------------- /examples/onnxruntime/quantization/text-classification/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/quantization/text-classification/requirements.txt -------------------------------------------------------------------------------- /examples/onnxruntime/quantization/text-classification/run_glue.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/quantization/text-classification/run_glue.py -------------------------------------------------------------------------------- /examples/onnxruntime/quantization/token-classification/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/quantization/token-classification/README.md -------------------------------------------------------------------------------- /examples/onnxruntime/quantization/token-classification/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/quantization/token-classification/requirements.txt -------------------------------------------------------------------------------- /examples/onnxruntime/quantization/token-classification/run_ner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/examples/onnxruntime/quantization/token-classification/run_ner.py -------------------------------------------------------------------------------- /optimum/commands/export/onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/commands/export/onnx.py -------------------------------------------------------------------------------- /optimum/commands/onnxruntime/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/commands/onnxruntime/base.py -------------------------------------------------------------------------------- /optimum/commands/onnxruntime/optimize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/commands/onnxruntime/optimize.py -------------------------------------------------------------------------------- /optimum/commands/onnxruntime/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/commands/onnxruntime/quantize.py -------------------------------------------------------------------------------- /optimum/commands/register/register_export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/commands/register/register_export.py -------------------------------------------------------------------------------- /optimum/commands/register/register_onnxruntime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/commands/register/register_onnxruntime.py -------------------------------------------------------------------------------- /optimum/exporters/onnx/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/exporters/onnx/__init__.py -------------------------------------------------------------------------------- /optimum/exporters/onnx/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/exporters/onnx/__main__.py -------------------------------------------------------------------------------- /optimum/exporters/onnx/_traceable_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/exporters/onnx/_traceable_cache.py -------------------------------------------------------------------------------- /optimum/exporters/onnx/_traceable_decorator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/exporters/onnx/_traceable_decorator.py -------------------------------------------------------------------------------- /optimum/exporters/onnx/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/exporters/onnx/base.py -------------------------------------------------------------------------------- /optimum/exporters/onnx/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/exporters/onnx/config.py -------------------------------------------------------------------------------- /optimum/exporters/onnx/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/exporters/onnx/constants.py -------------------------------------------------------------------------------- /optimum/exporters/onnx/convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/exporters/onnx/convert.py -------------------------------------------------------------------------------- /optimum/exporters/onnx/input_generators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/exporters/onnx/input_generators.py -------------------------------------------------------------------------------- /optimum/exporters/onnx/model_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/exporters/onnx/model_configs.py -------------------------------------------------------------------------------- /optimum/exporters/onnx/model_patcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/exporters/onnx/model_patcher.py -------------------------------------------------------------------------------- /optimum/exporters/onnx/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/exporters/onnx/utils.py -------------------------------------------------------------------------------- /optimum/onnx/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnx/__init__.py -------------------------------------------------------------------------------- /optimum/onnx/graph_transformations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnx/graph_transformations.py -------------------------------------------------------------------------------- /optimum/onnx/transformations_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnx/transformations_utils.py -------------------------------------------------------------------------------- /optimum/onnx/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnx/utils.py -------------------------------------------------------------------------------- /optimum/onnx/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnx/version.py -------------------------------------------------------------------------------- /optimum/onnxruntime/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnxruntime/__init__.py -------------------------------------------------------------------------------- /optimum/onnxruntime/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnxruntime/base.py -------------------------------------------------------------------------------- /optimum/onnxruntime/configuration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnxruntime/configuration.py -------------------------------------------------------------------------------- /optimum/onnxruntime/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnxruntime/constants.py -------------------------------------------------------------------------------- /optimum/onnxruntime/dummy_objects.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnxruntime/dummy_objects.py -------------------------------------------------------------------------------- /optimum/onnxruntime/modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnxruntime/modeling.py -------------------------------------------------------------------------------- /optimum/onnxruntime/modeling_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnxruntime/modeling_decoder.py -------------------------------------------------------------------------------- /optimum/onnxruntime/modeling_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnxruntime/modeling_diffusion.py -------------------------------------------------------------------------------- /optimum/onnxruntime/modeling_ort.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnxruntime/modeling_ort.py -------------------------------------------------------------------------------- /optimum/onnxruntime/modeling_seq2seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnxruntime/modeling_seq2seq.py -------------------------------------------------------------------------------- /optimum/onnxruntime/optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnxruntime/optimization.py -------------------------------------------------------------------------------- /optimum/onnxruntime/pipelines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnxruntime/pipelines.py -------------------------------------------------------------------------------- /optimum/onnxruntime/preprocessors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnxruntime/preprocessors/__init__.py -------------------------------------------------------------------------------- /optimum/onnxruntime/preprocessors/passes/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnxruntime/preprocessors/passes/__init__.py -------------------------------------------------------------------------------- /optimum/onnxruntime/preprocessors/passes/excluders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnxruntime/preprocessors/passes/excluders.py -------------------------------------------------------------------------------- /optimum/onnxruntime/preprocessors/passes/fully_connected.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnxruntime/preprocessors/passes/fully_connected.py -------------------------------------------------------------------------------- /optimum/onnxruntime/preprocessors/passes/gelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnxruntime/preprocessors/passes/gelu.py -------------------------------------------------------------------------------- /optimum/onnxruntime/preprocessors/passes/layernorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnxruntime/preprocessors/passes/layernorm.py -------------------------------------------------------------------------------- /optimum/onnxruntime/preprocessors/quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnxruntime/preprocessors/quantization.py -------------------------------------------------------------------------------- /optimum/onnxruntime/quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnxruntime/quantization.py -------------------------------------------------------------------------------- /optimum/onnxruntime/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/optimum/onnxruntime/utils.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/pyproject.toml -------------------------------------------------------------------------------- /tests/assets/onnx/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/tests/assets/onnx/config.json -------------------------------------------------------------------------------- /tests/assets/onnx/model.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/tests/assets/onnx/model.onnx -------------------------------------------------------------------------------- /tests/cli/test_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/tests/cli/test_cli.py -------------------------------------------------------------------------------- /tests/exporters/common/test_tasks_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/tests/exporters/common/test_tasks_manager.py -------------------------------------------------------------------------------- /tests/exporters/onnx/test_export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/tests/exporters/onnx/test_export.py -------------------------------------------------------------------------------- /tests/exporters/onnx/test_export_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/tests/exporters/onnx/test_export_cli.py -------------------------------------------------------------------------------- /tests/exporters/onnx/utils_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/tests/exporters/onnx/utils_tests.py -------------------------------------------------------------------------------- /tests/onnx/test_onnx_export_custom_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/tests/onnx/test_onnx_export_custom_module.py -------------------------------------------------------------------------------- /tests/onnx/test_onnx_graph_transformations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/tests/onnx/test_onnx_graph_transformations.py -------------------------------------------------------------------------------- /tests/onnxruntime/test_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/tests/onnxruntime/test_decoder.py -------------------------------------------------------------------------------- /tests/onnxruntime/test_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/tests/onnxruntime/test_diffusion.py -------------------------------------------------------------------------------- /tests/onnxruntime/test_modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/tests/onnxruntime/test_modeling.py -------------------------------------------------------------------------------- /tests/onnxruntime/test_optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/tests/onnxruntime/test_optimization.py -------------------------------------------------------------------------------- /tests/onnxruntime/test_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/tests/onnxruntime/test_quantization.py -------------------------------------------------------------------------------- /tests/onnxruntime/test_seq2seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/tests/onnxruntime/test_seq2seq.py -------------------------------------------------------------------------------- /tests/onnxruntime/test_timm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/tests/onnxruntime/test_timm.py -------------------------------------------------------------------------------- /tests/onnxruntime/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/tests/onnxruntime/test_utils.py -------------------------------------------------------------------------------- /tests/onnxruntime/testing_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huggingface/optimum-onnx/HEAD/tests/onnxruntime/testing_utils.py --------------------------------------------------------------------------------