Didier commited on
Commit
e79ab5a
·
verified ·
1 Parent(s): a9292a1

Update model_llm.py

Browse files
Files changed (1) hide show
  1. model_llm.py +1 -1
model_llm.py CHANGED
@@ -14,7 +14,7 @@ from transformers import TextIteratorStreamer
14
  from transformers import BitsAndBytesConfig
15
  from threading import Thread
16
 
17
- model_name = "mistralai/Mistral-7B-Instruct-v0.2"
18
 
19
  # Auto-regressive model for language completion: padding left
20
  tokenizer = AutoTokenizer.from_pretrained(model_name, padding_side="left")
 
14
  from transformers import BitsAndBytesConfig
15
  from threading import Thread
16
 
17
+ model_name = "mistralai/Mistral-7B-Instruct-v0.3"
18
 
19
  # Auto-regressive model for language completion: padding left
20
  tokenizer = AutoTokenizer.from_pretrained(model_name, padding_side="left")