You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Baizhou Zhang c5fd4aa6e8 [lora] add lora APIs for booster, support lora for TorchDDP (#4981) ... * add apis and peft requirement * add liscense and implement apis * add checkpointio apis * add torchddp fwd_bwd test * add support_lora methods * add checkpointio test and debug * delete unneeded codes * remove peft from LICENSE * add concrete methods for enable_lora * simplify enable_lora api * fix requirements		1 year ago
..
__init__.py	[misc] update pre-commit and run all files (#4752)	1 year ago
checkpoint_io_base.py	[lora] add lora APIs for booster, support lora for TorchDDP (#4981)	1 year ago
general_checkpoint_io.py	[lora] add lora APIs for booster, support lora for TorchDDP (#4981)	1 year ago
hybrid_parallel_checkpoint_io.py	[checkpointio] support unsharded checkpointIO for hybrid parallel (#4774)	1 year ago
index_file.py	[misc] update pre-commit and run all files (#4752)	1 year ago
utils.py	[checkpointio] hotfix torch 2.0 compatibility (#4824)	1 year ago