optimum
7f62e7d1 - Automatic framework detection in TasksManager for large models (#883)

Comment changes are shownComment changes are hidden
Commit
2 years ago
Automatic framework detection in TasksManager for large models (#883) * fix framework detection for large files * Update optimum/exporters/tasks.py Co-authored-by: Michael Benayoun <mickbenayoun@gmail.com> * fix on suggestion * avoid using generator as variable can't be reused * fix test * fix subfolder --------- Co-authored-by: Michael Benayoun <mickbenayoun@gmail.com>
Author
Parents
  • optimum
    • exporters
      • File
        tasks.py
    • onnxruntime
      • File
        modeling_diffusion.py