ubiodee commited on
Commit
6cdf960
Β·
verified Β·
1 Parent(s): b0dad4b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -19,9 +19,9 @@ def generate_response(prompt):
19
  with torch.no_grad():
20
  outputs = model.generate(
21
  **inputs,
22
- max_new_tokens=500,
23
- temperature=0.7,
24
- top_p=0.9,
25
  do_sample=True,
26
  eos_token_id=tokenizer.eos_token_id,
27
  pad_token_id=tokenizer.pad_token_id,
 
19
  with torch.no_grad():
20
  outputs = model.generate(
21
  **inputs,
22
+ max_new_tokens=250,
23
+ temperature=0.4,
24
+ top_p=0.4,
25
  do_sample=True,
26
  eos_token_id=tokenizer.eos_token_id,
27
  pad_token_id=tokenizer.pad_token_id,