ITRF
Collection
Instruction tuning a model on RAG and then finetuning a model on RAG data and then a reranker on the transition probabilities (Thesis weights). • 3 items • Updated
How to use tristanratz/itrf-reranker with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="tristanratz/itrf-reranker") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("tristanratz/itrf-reranker")
model = AutoModelForSequenceClassification.from_pretrained("tristanratz/itrf-reranker", device_map="auto")