navid72m commited on
Commit
63b8e0a
·
1 Parent(s): 5082ba6
Files changed (1) hide show
  1. 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 = "misteralai/mistral_7b_en" # Replace with the appropriate Mistral model name if available
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