akjindal53244 commited on
Commit
c863a93
1 Parent(s): e2858b5

Add model details in README.md

Browse files
Files changed (1) hide show
  1. README.md +50 -0
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)