Welcome to the LLaMA Math Wizard project! We're teaching an AI llama to be a math genius. ๐
Ever wished you had a magical llama that could solve math problems? Well, now you do! We've taken the brilliant LLaMA-2-7b-chat model and sent it to math school. The result? A fuzzy, four-legged, formula-crunching friend!
- ๐ง Fine-tuned LLaMA-2-7b-chat model with a PhD in Mathematics
- ๐ Trained on a diverse mathematics dataset
- ๐ฅ๏ธ Streamlit app for chatting with your new math tutor
- ๐ TensorBoard integration for beautiful learning curves
- ๐ QLoRA magic for efficient training
- Clone this repository (no llamas were harmed in the process):
git clone https://github.com/utkarshpophli/llama-math-finetuning.git
cd llama-math-wizard
- Install the required packages (llama food):
pip install -r requirements.txt
- Log in to Hugging Face (llama face optional):
huggingface-cli login
- Train your llama:
python src/train.py
- Chat with your math wizard:
streamlit run app.py
src/
: The llama's brain ๐งapp/
: The llama's mouth ๐ฃ๏ธnotebooks/
: The llama's sandbox ๐๏ธlogs/
: The llama's diary ๐
Got ideas to make our llama even smarter? We're all ears (and wool)! Feel free to open an issue or submit a pull request.
This project is licensed under the MIT License - see the LICENSE file for details.
- The incredible Hugging Face ๐ค team
- The LAION AI team for blessing us with LLaMA
- Caffeine โ, for fueling late-night coding sessions
Remember: Be kind to llamas, they're doing their best to understand math! ๐ฆโค๏ธ๐ข