Felladrin commited on
Commit
2b62c2d
1 Parent(s): 4d3e62a

Add Felladrin/Minueza-32M-UltraChat model

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -33,12 +33,13 @@ def generate(
33
  elif model_name == "Felladrin/TinyMistral-248M-SFT-v4":
34
  outputs = pipe(prompt, max_length=1024, use_cache=True, penalty_alpha=0.5, top_k=5, repetition_penalty=1.001)
35
  else:
36
- outputs = pipe(prompt, max_length=1024, do_sample=True, temperature=0.7, top_k=35, top_p=0.5, repetition_penalty=1.176)
37
 
38
  return outputs[0]["generated_text"]
39
 
40
  model_choices = [
41
  "Felladrin/Llama-160M-Chat-v1",
 
42
  "Felladrin/Minueza-32M-Chat",
43
  "Felladrin/Smol-Llama-101M-Chat-v1",
44
  "Felladrin/TinyMistral-248M-SFT-v4",
 
33
  elif model_name == "Felladrin/TinyMistral-248M-SFT-v4":
34
  outputs = pipe(prompt, max_length=1024, use_cache=True, penalty_alpha=0.5, top_k=5, repetition_penalty=1.001)
35
  else:
36
+ outputs = pipe(prompt, max_length=1024, do_sample=True, temperature=0.65, top_k=35, top_p=0.55, repetition_penalty=1.176)
37
 
38
  return outputs[0]["generated_text"]
39
 
40
  model_choices = [
41
  "Felladrin/Llama-160M-Chat-v1",
42
+ "Felladrin/Minueza-32M-UltraChat",
43
  "Felladrin/Minueza-32M-Chat",
44
  "Felladrin/Smol-Llama-101M-Chat-v1",
45
  "Felladrin/TinyMistral-248M-SFT-v4",