thon | |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM | |
import torch | |
tokenizer = AutoTokenizer.from_pretrained("google/byt5-base") | |
model = AutoModelForSeq2SeqLM.from_pretrained("google/byt5-base") | |
input_ids_prompt = "The dog chases a ball in the park." |