fuel-examples 0.20.0

Minimalist ML framework.
1
2
3
4
5
6
from transformers import AutoModelForCausalLM, AutoTokenizer

BASE_MODEL = "google/t5-v1_1-xxl"
tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
# The tokenizer will be saved in /tmp/tokenizer/tokenizer.json
tokenizer.save_pretrained("/tmp/tokenizer/")