Spaces:

pvyas96
/

mistral_streamlit_chat

Sleeping

pvyas96 commited on May 11

Commit

a600542

•

1 Parent(s): d9a9459

Update pages/1_Simple_Chat_UI.py

Files changed (1) hide show

pages/1_Simple_Chat_UI.py CHANGED Viewed

@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForSeq2Se
 def load_model_tokenizer(model_name, hf_api_key):
     if model_name == "Mistral-7B":
-        model_name="mistralai/Mistral-7B-Instruct-v0.2"
         model = AutoModelForCausalLM.from_pretrained(model_name, token=hf_api_key)
         tokenizer = AutoTokenizer.from_pretrained(model_name, tokenizer=hf_api_key)
     elif model_name == "blenderbot-400M-distill":

 def load_model_tokenizer(model_name, hf_api_key):
     if model_name == "Mistral-7B":
+        model_name="NousResearch/Nous-Hermes-2-Mistral-7B-DPO"
         model = AutoModelForCausalLM.from_pretrained(model_name, token=hf_api_key)
         tokenizer = AutoTokenizer.from_pretrained(model_name, tokenizer=hf_api_key)
     elif model_name == "blenderbot-400M-distill":