How to use google/switch-base-256 with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("google/switch-base-256") model = AutoModelForSeq2SeqLM.from_pretrained("google/switch-base-256", device_map="auto")
https://huggingface.co/google/switch-base-256/resolve/refs%2Fpr%2F7/pytorch_model.bin.index.json
hf download hf://google/switch-base-256@refs/pr/7/pytorch_model.bin.index.json
curl -L -o pytorch_model.bin.index.json https://huggingface.co/google/switch-base-256/resolve/refs%2Fpr%2F7/pytorch_model.bin.index.json
