| library_name: peft | |
| base_model: mistralai/Mistral-7B-v0.1 | |
| pipeline_tag: text-generation | |
| 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 3*3=<<3*3=9>>9 sprints a week | |
| He runs 60*9=<<60*9=540>>540 meters a week | |
| #### 540\ | |
| ---\ | |
| 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) | |
| ``` |