ColossalAI/tests/test_lora
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
2023-10-31 15:19:37 +08:00
..
test_torch_ddp_lora.py [lora] add lora APIs for booster, support lora for TorchDDP (#4981) 2023-10-31 15:19:37 +08:00