Files
candle/candle-examples/examples/flux/t5_tokenizer.py
2024-08-05 21:03:31 +02:00

7 lines
258 B
Python

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/")