Spaces:

aifeifei798
/

Duke86Syl

Running on Zero

aifeifei798 commited on Apr 2

Commit

918cd57

verified ·

1 Parent(s): e728a98

Update feifeilib/feifeichat.py

Files changed (1) hide show

feifeilib/feifeichat.py CHANGED Viewed

@@ -174,7 +174,7 @@ def feifeiimgprompt(message_files, message_text, image_mod):
 def feifeichatmod(additional_dropdown, input_prompt):
     if additional_dropdown == "mistralai/Mistral-Nemo-Instruct-2411":
-        model = "mistral-large-2411"
         stream_response = Mistralclient.chat.stream(model=model, messages=input_prompt)
         partial_message = ""
         for chunk in stream_response:

 def feifeichatmod(additional_dropdown, input_prompt):
     if additional_dropdown == "mistralai/Mistral-Nemo-Instruct-2411":
+        model = "mistral-small-latest"
         stream_response = Mistralclient.chat.stream(model=model, messages=input_prompt)
         partial_message = ""
         for chunk in stream_response: