optimum
50531a42 - Adding Onnx Support For Megatron-Bert (#2169)

Comment changes are shownComment changes are hidden
Commit
142 days ago
Adding Onnx Support For Megatron-Bert (#2169) * Adding Onnx Support For Megatron-Bert * fix style * fix style * fix style --------- Co-authored-by: pragyan <pragyan.singh@tarkas.ai> Co-authored-by: Ella Charlaix <80481427+echarlaix@users.noreply.github.com>
Author
Parents
  • optimum/exporters
    • onnx
      • File
        model_configs.py
    • File
      tasks.py
  • tests/exporters
    • File
      exporters_utils.py