schuler commited on
Commit
45d4d15
·
verified ·
1 Parent(s): ff32de9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -34,10 +34,10 @@ def local_generate(
34
  prompt,
35
  generation_config,
36
  max_new_tokens,
37
- do_sample,
38
- top_p,
39
- repetition_penalty,
40
- temperature=temperature
41
  ):
42
  response_output = generator(
43
  prompt,
 
34
  prompt,
35
  generation_config,
36
  max_new_tokens,
37
+ do_sample=True,
38
+ top_p=0.25,
39
+ repetition_penalty=1.2,
40
+ temperature=1.0
41
  ):
42
  response_output = generator(
43
  prompt,