How to use lgrobol/electra-minuscule-discriminator with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="lgrobol/electra-minuscule-discriminator")
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForTokenClassification tokenizer = AutoTokenizer.from_pretrained("lgrobol/electra-minuscule-discriminator") model = AutoModelForTokenClassification.from_pretrained("lgrobol/electra-minuscule-discriminator", device_map="auto")
A ridiculously small ELECTRA discriminator model for testing purposes.
THIS MODEL HAS NOT BEEN TRAINED, DO NOT EXPECT ANYThING OF IT.
Files info
