Mistral 7B RAG Instruct

Published by @ragsmith · Community adapter

Optimised for retrieval-augmented generation pipelines. Trained to faithfully cite provided context chunks, refuse when context is insufficient, and format answers with inline references.

Usage

from peft import PeftModel
from transformers import AutoModelForCausalLM, AutoTokenizer

tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.3")
base = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.3")
model = PeftModel.from_pretrained(base, "modelforgelab/mistral-7b-rag-instruct-lora")

inputs = tokenizer("Your prompt here", return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=256)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))

Compatibility

• transformers>=4.38
• vLLM
• LangChain