Tiny LM
Collection
19 items โข Updated โข 2
# Load model directly
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("trl-internal-testing/tiny-BartModel")
model = AutoModel.from_pretrained("trl-internal-testing/tiny-BartModel")This is a minimal model built for unit tests in the TRL library.
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="trl-internal-testing/tiny-BartModel")