Aaresh5308a commited on
Commit
c1d6742
·
1 Parent(s): 4b9b37d
Files changed (1) hide show
  1. app.py +10 -4
app.py CHANGED
@@ -48,11 +48,17 @@ def generate(req: Request):
48
  outputs = model.generate(
49
  **inputs,
50
  max_new_tokens=req.max_tokens,
 
51
  do_sample=True,
52
- temperature=req.temperature,
53
- top_p=0.85, # stability fix
54
- top_k=40, # stability fix
55
- repetition_penalty=1.1 # prevents loops
 
 
 
 
 
56
  )
57
 
58
  text = tokenizer.decode(
 
48
  outputs = model.generate(
49
  **inputs,
50
  max_new_tokens=req.max_tokens,
51
+
52
  do_sample=True,
53
+ temperature=0.8,
54
+ top_p=0.92,
55
+ top_k=40,
56
+
57
+ repetition_penalty=1.7, # strong anti-loop
58
+ no_repeat_ngram_size=5, # stronger phrase blocking
59
+
60
+ eos_token_id=tokenizer.eos_token_id,
61
+ pad_token_id=tokenizer.eos_token_id
62
  )
63
 
64
  text = tokenizer.decode(