Downloads · 30 days
7
9% of all-time downloads
mmoffatt/custom_gpt2
custom_gpt2 is a machine learning model from mmoffatt. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
This is a custom implementation of gpt2, where we replace attention with our implementation. Currently, we don't replace softmax, but in future submits we would like to replace the softmax function in attention with o…
Downloads · 30 days
7
9% of all-time downloads
All-time downloads
78
Public
Parameters
124M
996 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors498 MB · 99%
From the Hugging Face model README
This is a custom implementation of gpt2, where we replace attention with our implementation. Currently, we don't replace softmax, but in future submits we would like to replace the softmax function in attention with other softmax variations.
We directly use the huggingface gpt2 model: https://huggingface.co/openai-community/gpt2
This model was finetuned on the wikitext dataset: https://paperswithcode.com/dataset/wikitext-2
base model: huggingface gpt2