Update parameters for Felladrin/TinyMistral-248M-Chat-v2

app.py

@@ -27,8 +27,8 @@ def generate(
         outputs = pipe(prompt, max_new_tokens=250, use_cache=True, penalty_alpha=0.5, top_k=4, repetition_penalty=1.105)
     elif model_name == "Felladrin/Llama-160M-Chat-v1":
         outputs = pipe(prompt, max_new_tokens=250, use_cache=True, penalty_alpha=0.5, top_k=4, repetition_penalty=1.01)
-    elif model_name == "Felladrin/TinyMistral-248M-Chat-v1":
-        outputs = pipe(prompt, max_new_tokens=250, use_cache=True, penalty_alpha=0.45, top_k=4, repetition_penalty=1.03, guidance_scale=1.3)
+    elif model_name == "Felladrin/TinyMistral-248M-Chat-v2":
+        outputs = pipe(prompt, max_new_tokens=250, use_cache=True, penalty_alpha=0.5, top_k=5, repetition_penalty=1.0)
     else:
         outputs = pipe(prompt, max_new_tokens=250, do_sample=True, temperature=0.65, top_k=35, top_p=0.55, repetition_penalty=1.176)
         
@@ -37,7 +37,7 @@ def generate(
 model_choices = [
     "Felladrin/Llama-160M-Chat-v1",
     "Felladrin/Minueza-32Mx2-Chat",
-    "Felladrin/TinyMistral-248M-Chat-v1",
+    "Felladrin/TinyMistral-248M-Chat-v2",
     "Felladrin/Llama-68M-Chat-v1",
     "Felladrin/Minueza-32M-UltraChat",
     "Felladrin/Minueza-32M-Deita",