Downloads · 30 days
18
5% of all-time downloads
bioelectra-small-generator
molly-hayward/bioelectra-small-generator
bioelectra-small-generator is a machine learning model from molly-hayward. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers.
To produce BioELECTRA, we pretrain ELECTRA on a corpus of over 20 million abstracts from PubMed.
All-time downloads
364
Public
Repo size
55.4 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.h518.6 MB · 50%
At a glance
- Library
- transformers
- Model type
- electra
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Apr 17, 2021
- SHA
- 336833d8
Model card
From the Hugging Face model README
To produce BioELECTRA, we pretrain ELECTRA on a corpus of over 20 million abstracts from PubMed.
How to use the generator in transformers:
from transformers import ElectraForMaskedLM, ElectraTokenizerFast
import torch
generator = ElectraForMaskedLM.from_pretrained("molly-hayward/bioelectra-small-generator")
tokenizer = ElectraTokenizerFast.from_pretrained("molly-hayward/bioelectra-small-generator")
- Library
- transformers
- Type
- electra
- Created
- Mar 2, 2022
- Updated
- Apr 17, 2021