Training in progress, step 800 57d839e
Department of Computational Linguistics - University of Trier commited on
How to use cl-trier/flan-t5-base_twon-generative-agent with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("cl-trier/flan-t5-base_twon-generative-agent")
model = AutoModelForSeq2SeqLM.from_pretrained("cl-trier/flan-t5-base_twon-generative-agent", device_map="auto")