Lightning-AI / Lightning-AI/litgpt
Request to support OLMoE
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/allenai/OLMoE is a very good MOE structure, with 7B parameter and 1B activate parameter. It is very suitable for academic research. Whether the litgpt team considers to support this model?
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
Start with the linked OLMoE repository and compare its model structure with LitGPT’s existing model-support entry points. No LitGPT files or tests are named, so first define the integration scope and validation needed; done means OLMoE support requirements are concrete and the model is supported by LitGPT.
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
- 25/100