Views
No views yet
letxbe/mistral-7b-v03-BoundingDocs-rephrased is a fine-tuned Mistral-7B-v0.3 for the Document Question Answering task. It was trained on BoundingDocs using the rephrased version of the questions.Mistral-7B-v0.31
2TEMPLATE_PROMPT = '''<|startdocument|>
3{DOCUMENT}
4<|enddocument|>
5
6<|starttask|>
7Answer the following question about the document:
8Question: "{QUESTION}"
9Answer completing the following format:
10
11'''json
12{"value": ""}
13'''
14
15<|endtask|>
16'''1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3# Load model and tokenizer
4tokenizer = AutoTokenizer.from_pretrained("letxbe/mistral-7b-v03-BoundingDocs-rephrased")
5model = AutoModelForCausalLM.from_pretrained("letxbe/mistral-7b-v03-BoundingDocs-rephrased")
6
7# Encode input
8input_text = "Your prompt"
9inputs = tokenizer(input_text, return_tensors="pt")
10
11# Generate response
12outputs = model.generate(**inputs)
13
14# Decode and print
15print(tokenizer.decode(outputs[0], skip_special_tokens=True))