ColossalAI icon indicating copy to clipboard operation
ColossalAI copied to clipboard

[FEATURE]: LoRA with sharded model

Open KaiLv69 opened this issue 1 year ago • 1 comments

Describe the feature

Hi, when training big model like llama2-70b with lora, it will run into oom due to the unsharded model.

It could help a lot if lora supported with GeminiPlugin or HybridParallelPlugin. Wonder if any plan to support that?

KaiLv69 avatar Jun 17 '24 11:06 KaiLv69

Hi, the HybridParallelPlugin has already support lora strategy. We would appreciate your feedback.

wangbluo avatar Aug 26 '24 06:08 wangbluo