Edit model card

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
27
Safetensors
Model size
67M params
Tensor type
F32
·
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Dataset used to train sms1097/retrieval_model

Collection including sms1097/retrieval_model