Spaces:
Runtime error
Runtime error
misteral
Browse files- selection.py +1 -1
selection.py
CHANGED
|
@@ -34,7 +34,7 @@ most_relevant_context = find_most_relevant_context(contexts, Question)
|
|
| 34 |
instruction = most_relevant_context[:300] + " " + Question
|
| 35 |
|
| 36 |
# Load Mistral model and tokenizer
|
| 37 |
-
model_name = "
|
| 38 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 39 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
| 40 |
|
|
|
|
| 34 |
instruction = most_relevant_context[:300] + " " + Question
|
| 35 |
|
| 36 |
# Load Mistral model and tokenizer
|
| 37 |
+
model_name = "mistralai/Mistral-7B-v0.1" # Replace with the appropriate Mistral model name if available
|
| 38 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 39 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
| 40 |
|