Hugging Face's logo

sms1097
/
retrieval_model

Text Classification
Transformers
TensorBoard
Safetensors
distilbert
text-embeddings-inference
Model card Files
xet
 Metrics Community
How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-classification", model="sms1097/retrieval_model")
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("sms1097/retrieval_model")
model = AutoModelForSequenceClassification.from_pretrained("sms1097/retrieval_model")
Quick Links

Retrieval Model

This generates the Retrieve token as descirbed in Self-RAG.

We are testing to see if a retrieved document is relevant to the user input of our language model.

The expected input to the model is just a query posed to the language model.

Downloads last month
7
Safetensors
Model size
67M params
Tensor type
F32
·
Inference Providers NEW
Text Classification
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Dataset used to train sms1097/retrieval_model

Collection including sms1097/retrieval_model