How to use google/switch-base-64 with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("google/switch-base-64") model = AutoModelForSeq2SeqLM.from_pretrained("google/switch-base-64")
Fixes https://huggingface.co/google/switch-base-32/discussions/5#63b370847091e602f1a210eb
· Sign up or log in to comment
