Downloads · 30 days
18
2% of all-time downloads
gpt-neo-math-small
Saisam/gpt-neo-math-small
gpt-neo-math-small is a text generation model from Saisam. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as apache-2.0.
In the project, we used an HuggingFace GPT-NEO small model and fine-tuned the tactic dataset. The Input should be of the form
All-time downloads
1.1K
Public
Repo size
1.1 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin551 MB · 100%
At a glance
- Task
- Text Generation
- Library
- transformers
- License
- apache-2.0
- Model type
- gpt_neo
- Access
- Public
- Created
- Apr 21, 2022
- Updated
- Dec 26, 2022
- SHA
- d2392642
Try a prompt
Model card
From the Hugging Face model README
GPT-NEO-Model for Lean Tactics
In the project, we used an HuggingFace GPT-NEO small model and fine-tuned the tactic dataset. The Input should be of the form
<GOAL> Goal <PROOFSTEP>
The model can easily be accessed using the following code.
from transformers import GPT2Tokenizer, GPTNeoForCausalLM
import torch
tokenizer = GPT2Tokenizer.from_pretrained("Saisam/gpt-neo-math-small")
model = GPTNeoForCausalLM.from_pretrained("Saisam/gpt-neo-math-small")
More Information can be found at https://github.com/saisurbehera/mathProof.
The current model beats the GPT-F for minif2f benchmark
Worked along with Xihao Xhang and Moya Zhu
- Task
- Text Generation
- Library
- transformers
- Type
- gpt_neo
- License
- apache-2.0
- Created
- Apr 21, 2022
- Updated
- Dec 26, 2022