Instructions to use sms1097/retrieval_model with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use sms1097/retrieval_model with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="sms1097/retrieval_model")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("sms1097/retrieval_model") model = AutoModelForSequenceClassification.from_pretrained("sms1097/retrieval_model", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Create README.md
Browse files
README.md
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
datasets:
|
| 4 |
+
- sms1097/self_rag_tokens_train_data
|
| 5 |
+
---
|
| 6 |
+
|
| 7 |
+
# Retrieval Prediction Model
|
| 8 |
+
Classification model for an instruction model to determine if it requires additional context to answer a question.
|