Downloads · 30 days
704
1% of all-time downloads
predibase/gsm8k
gsm8k is a text generation model from predibase. Use it when you need the model to write or continue text. It is set up for peft.
Description: Grade school math problems\ Original dataset: https://huggingface.co/datasets/gsm8k \ ---\ Try querying this adapter for free in Lora Land at https://predibase.com/lora-land! \ The adaptercategory is STEM…
Downloads · 30 days
704
1% of all-time downloads
All-time downloads
94.1K
Public
Repo size
13.6 MB
Likes
4
Public
Click a slice to open those files.
.safetensors13.6 MB · 100%
From the Hugging Face model README
Description: Grade school math problems
Original dataset: https://huggingface.co/datasets/gsm8k
---
Try querying this adapter for free in Lora Land at https://predibase.com/lora-land!
The adapter_category is STEM and the name is Grade School Math (gsm8k)
---
Sample input: Please answer the following question: James decides to run 3 sprints 3 times a week. He runs 60 meters each sprint. How many total meters does he run a week?\nAnswer:
---
Sample output: He runs 33=<<33=9>>9 sprints a week
He runs 609=<<609=540>>540 meters a week
---
Try using this adapter yourself!
from transformers import AutoModelForCausalLM, AutoTokenizer
model_id = "mistralai/Mistral-7B-v0.1"
peft_model_id = "predibase/gsm8k"
model = AutoModelForCausalLM.from_pretrained(model_id)
model.load_adapter(peft_model_id)