Lightning-AI / Lightning-AI/litgpt
Feature request: mezo error propagation?
Open
Nobody has claimed this yet.
enhancement
- Dominant language
- Python
- Stars
- 13.7k
- Forks
- 1.5k
- Avg merge
- 15h 37m
- Merged PRs (30d)
- 1
Description
https://github.com/princeton-nlp/MeZO
I'm wondering how feasible it is to incorporate mezo into this boilerplate code to enable training llms on much smaller footprint of vram.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
The issue names no LitGPT files or tests. Start by reviewing the linked MeZO project and LitGPT’s training entry points to determine the integration scope. Done would mean a defined and working MeZO option for lower-VRAM LLM training, with appropriate coverage for its training path.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- ai, machine-learning
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 20/100