Add electra generator model 701d7dc
aapot commited on
How to use Finnish-NLP/electra-base-generator-finnish with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("fill-mask", model="Finnish-NLP/electra-base-generator-finnish") # Load model directly
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("Finnish-NLP/electra-base-generator-finnish")
model = AutoModelForMaskedLM.from_pretrained("Finnish-NLP/electra-base-generator-finnish", device_map="auto")