optimum
66912e79 - Support customize dtype for dummy generators (#1307)

Commit

1 year ago

Support customize dtype for dummy generators (#1307) * add dtype customize * fix * fix key * adapt exporter * fix exporter test * fix cli test * fix cli test * address review * address review * fix typo * update script to the convention

References

#1307 - Support customize dtype for dummy generators

Author

JingyaHuang

Parents

234a4274

Files10

optimum
- exporters/onnx
  - __main__.py
  - base.py
  - config.py
  - convert.py
  - model_configs.py
  - utils.py
- utils
  - __init__.py
  - input_generators.py
tests
- exporters/onnx
  - test_onnx_export.py
- utils
  - test_dummpy_input_generators.py

optimum 66912e79 - Support customize dtype for dummy generators (#1307)

optimum
66912e79 - Support customize dtype for dummy generators (#1307)