AnswerDotAI / AnswerDotAI/fsdp_qlora
Can i use this script to pre-train models?
Open
- Dominant language
- Jupyter Notebook
- Stars
- 1.6k
- Forks
- 201
- PR merge metrics
- No merged PRs in 30d
Description
Hi there!
I am currently working on pre-training a Llama or Mistral model with clinical texts. Is there any way to use this QLoRA + FSDP script to do such training? Should I make any changes to the current code to be able to do pre-training?
Contributor guide
No contributing guide indexed for this repository
Research direction
Start by reading the QLoRA + FSDP script and its documented training flow, then compare the stated pre-training goal for Llama or Mistral models with the current scope. Done would require a maintainer-confirmed path for pre-training on clinical texts and clear requirements for any code changes.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- jupyter-notebook
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 20/100