How to use junnyu/electra_small_discriminator with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForPreTraining tokenizer = AutoTokenizer.from_pretrained("junnyu/electra_small_discriminator") model = AutoModelForPreTraining.from_pretrained("junnyu/electra_small_discriminator")
SFconvertbot