aifeifei798 commited on
Commit
f0de52b
·
verified ·
1 Parent(s): fa145f8

Update feifeilib/feifeichat.py

Browse files
Files changed (1) hide show
  1. feifeilib/feifeichat.py +2 -1
feifeilib/feifeichat.py CHANGED
@@ -118,7 +118,8 @@ def feifeiimgprompt(message_files, message_text, image_mod):
118
 
119
  def feifeichatmod(additional_dropdown, input_prompt):
120
  if additional_dropdown == "mistralai/Mistral-Nemo-Instruct-2411":
121
- model = "mistral-large-2411"
 
122
  stream_response = Mistralclient.chat.stream(model=model, messages=input_prompt)
123
  partial_message = ""
124
  for chunk in stream_response:
 
118
 
119
  def feifeichatmod(additional_dropdown, input_prompt):
120
  if additional_dropdown == "mistralai/Mistral-Nemo-Instruct-2411":
121
+ #model = "mistral-large-2411"
122
+ model = "mistral-small-2501"
123
  stream_response = Mistralclient.chat.stream(model=model, messages=input_prompt)
124
  partial_message = ""
125
  for chunk in stream_response: