akjindal53244
commited on
Commit
•
c863a93
1
Parent(s):
e2858b5
Add model details in README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,53 @@
|
|
1 |
---
|
2 |
license: apache-2.0
|
|
|
|
|
|
|
|
|
3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
license: apache-2.0
|
3 |
+
language:
|
4 |
+
- en
|
5 |
+
tags:
|
6 |
+
- Mathematical Reasoning
|
7 |
---
|
8 |
+
# Model Card for Model ID
|
9 |
+
|
10 |
+
This model is instruction-tuned [Mistral-7B](https://huggingface.co/mistralai/Mistral-7B-v0.1) model using QLoRA on single RTX 4090 GPU. Model is tuned to reason and answer mathematical problems. Model is also capable of writing a Python program that upon compilation prints answer to the question. To generate python program, you can ask model to write a Python program (as part of prompt) along with question. Refer to **Results** section for examples.
|
11 |
+
|
12 |
+
## Model Details
|
13 |
+
|
14 |
+
It is a Instruction-tuned Mistral-7B and performs mathematical reasoning and optionally write a Python program.
|
15 |
+
|
16 |
+
### Model Description
|
17 |
+
|
18 |
+
- **Developed by:** [Ashvini Kumar Jindal](https://www.linkedin.com/in/ashvini-jindal-26653262/)
|
19 |
+
- **Funded by:** self-work
|
20 |
+
- **Model type:** Instruction-tuned
|
21 |
+
- **Language(s) (NLP):** English
|
22 |
+
- **Finetuned from model:** mistralai/Mistral-7B-v0.1
|
23 |
+
|
24 |
+
|
25 |
+
## Uses
|
26 |
+
|
27 |
+
Arithmo-Mistral-7B is trained with the following format:
|
28 |
+
|
29 |
+
```
|
30 |
+
Question: <question>
|
31 |
+
|
32 |
+
Answer:
|
33 |
+
```
|
34 |
+
|
35 |
+
It will perform best if queried in this way.
|
36 |
+
|
37 |
+
## How to Get Started with the Model
|
38 |
+
|
39 |
+
Model is compatibale with Huggingface. I will publish a generation/inference script soon. Model inference on CPU also works; I have tested it on Macbook M1 Pro. GPU inference is much faster than CPU inference.
|
40 |
+
|
41 |
+
|
42 |
+
### Results
|
43 |
+
|
44 |
+
Here are sample screenshots of model output for few questions :)
|
45 |
+
|
46 |
+
|
47 |
+
![image/png](https://cdn-uploads.huggingface.co/production/uploads/64c75c1237333ccfef30a602/qE0V8cZnvQDRIq6qANuYp.png)
|
48 |
+
|
49 |
+
|
50 |
+
![image/png](https://cdn-uploads.huggingface.co/production/uploads/64c75c1237333ccfef30a602/rXEzumBHG-y2HEhOhSRt2.png)
|
51 |
+
|
52 |
+
|
53 |
+
![image/png](https://cdn-uploads.huggingface.co/production/uploads/64c75c1237333ccfef30a602/X_hLjlNRBavb473ejgDIl.png)
|