How to use trl-internal-testing/tiny-T5ForConditionalGeneration with Transformers:
# 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")
This is a minimal model built for unit tests in the TRL library.
Files info
# 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")