Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[FEATURE]: LoRA with sharded model #5826

Open
KaiLv69 opened this issue Jun 17, 2024 · 0 comments
Open

[FEATURE]: LoRA with sharded model #5826

KaiLv69 opened this issue Jun 17, 2024 · 0 comments
Labels
enhancement New feature or request

Comments

@KaiLv69
Copy link
Contributor

KaiLv69 commented Jun 17, 2024

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 KaiLv69 added the enhancement New feature or request label Jun 17, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@KaiLv69