sablab commited on
Commit
5f2d498
·
verified ·
1 Parent(s): 4010b81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -10
app.py CHANGED
@@ -24,7 +24,7 @@ def generate(
24
 
25
  generate_kwargs = dict(
26
  temperature=temperature,
27
- max_new_tokens=8000,
28
  top_p=top_p,
29
  repetition_penalty=repetition_penalty,
30
  do_sample=True,
@@ -52,15 +52,6 @@ additional_inputs=[
52
  interactive=True,
53
  info="Higher values produce more diverse outputs",
54
  ),
55
- gr.Slider(
56
- label="Max new tokens",
57
- value=8000,
58
- minimum=4000,
59
- maximum=16000,
60
- step=64,
61
- interactive=True,
62
- info="The maximum numbers of new tokens",
63
- ),
64
  gr.Slider(
65
  label="Top-p (nucleus sampling)",
66
  value=0.90,
 
24
 
25
  generate_kwargs = dict(
26
  temperature=temperature,
27
+ max_new_tokens=4000,
28
  top_p=top_p,
29
  repetition_penalty=repetition_penalty,
30
  do_sample=True,
 
52
  interactive=True,
53
  info="Higher values produce more diverse outputs",
54
  ),
 
 
 
 
 
 
 
 
 
55
  gr.Slider(
56
  label="Top-p (nucleus sampling)",
57
  value=0.90,