How to use trl-internal-testing/tiny-T5ForConditionalGeneration with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("trl-internal-testing/tiny-T5ForConditionalGeneration") model = AutoModelForSeq2SeqLM.from_pretrained("trl-internal-testing/tiny-T5ForConditionalGeneration", device_map="auto")
https://huggingface.co/trl-internal-testing/tiny-T5ForConditionalGeneration/resolve/7fc635886b65864caa87b086109da2e220c902c8/tokenizer.json
hf download hf://trl-internal-testing/tiny-T5ForConditionalGeneration@7fc635886b65864caa87b086109da2e220c902c8/tokenizer.json
curl -L -o tokenizer.json https://huggingface.co/trl-internal-testing/tiny-T5ForConditionalGeneration/resolve/7fc635886b65864caa87b086109da2e220c902c8/tokenizer.json
