Ctaake commited on
Commit
d491e31
·
verified ·
1 Parent(s): e973e2c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -80,6 +80,8 @@ match checkpoint:
80
  format_prompt=format_prompt_nous
81
  case "mistralai/Mixtral-8x7B-Instruct-v0.1":
82
  format_prompt=format_prompt_mistral
 
 
83
 
84
  def inference(message, history, temperature=0.9, maxTokens=512, topP=0.9, repPenalty=1.1):
85
  # Updating the settings for the generation
 
80
  format_prompt=format_prompt_nous
81
  case "mistralai/Mixtral-8x7B-Instruct-v0.1":
82
  format_prompt=format_prompt_mistral
83
+ case "mistralai/Mistral-Nemo-Instruct-2407":
84
+ format_prompt=format_prompt_mistral
85
 
86
  def inference(message, history, temperature=0.9, maxTokens=512, topP=0.9, repPenalty=1.1):
87
  # Updating the settings for the generation