onnxruntime/orttraining/orttraining/python/training
Caroline Zhu 6236707c64
Enable >2GB models + allow model paths to be passed for generate_artifacts API (#20958)
### Description
Alternative design from #20942 

Allow users to pass in a model path for the generate_artifacts API. 

### Motivation and Context
- ONNX API calls such as the onnx checker + shape inference fail when
given a model > 2GB, but work if a path to a model >2GB is passed in.
2024-06-21 09:55:26 -07:00
..
amp [Better Engineering] Bump ruff to 0.0.278 and fix new lint errors (#16789) 2023-07-21 12:53:41 -07:00
api Introduce a Nominal Checkpoint for On-Device Training (#19232) 2024-01-30 22:11:25 -08:00
experimental Manage ORTModule configurations consistently (#16396) 2023-06-27 19:19:36 +08:00
onnxblock Enable >2GB models + allow model paths to be passed for generate_artifacts API (#20958) 2024-06-21 09:55:26 -07:00
optim Bump ruff to 0.3.2 and black to 24 (#19878) 2024-03-13 10:00:32 -07:00
ort_triton Support BFloat16 for Triton Codegen (#20353) 2024-04-18 17:15:11 +08:00
ortmodule Flash attention recompute (#20603) 2024-05-21 13:38:19 +08:00
utils Fix and enable few ORTModule Unit Tests (#19847) 2024-03-12 10:49:19 +08:00
__init__.py Bump ruff to 0.3.2 and black to 24 (#19878) 2024-03-13 10:00:32 -07:00
_utils.py Removed all the deprecated python training code and related tests and utils (#18333) 2023-11-17 18:19:21 -08:00
artifacts.py Enable >2GB models + allow model paths to be passed for generate_artifacts API (#20958) 2024-06-21 09:55:26 -07:00