dwb2023 commited on
Commit
ce7a117
·
verified ·
1 Parent(s): 421b69f

Update app.py

Browse files

tweak initial params

Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -13,7 +13,7 @@ from transformers import (
13
  )
14
 
15
  MAX_MAX_NEW_TOKENS = 1024
16
- DEFAULT_MAX_NEW_TOKENS = 256
17
  MAX_INPUT_TOKEN_LENGTH = 512
18
 
19
  DESCRIPTION = """\
@@ -104,14 +104,14 @@ chat_interface = gr.ChatInterface(
104
  minimum=0.05,
105
  maximum=1.0,
106
  step=0.05,
107
- value=0.4,
108
  ),
109
  gr.Slider(
110
  label="Top-k",
111
  minimum=1,
112
  maximum=1000,
113
  step=1,
114
- value=10,
115
  ),
116
  gr.Slider(
117
  label="Repetition penalty",
 
13
  )
14
 
15
  MAX_MAX_NEW_TOKENS = 1024
16
+ DEFAULT_MAX_NEW_TOKENS = 50
17
  MAX_INPUT_TOKEN_LENGTH = 512
18
 
19
  DESCRIPTION = """\
 
104
  minimum=0.05,
105
  maximum=1.0,
106
  step=0.05,
107
+ value=0.5,
108
  ),
109
  gr.Slider(
110
  label="Top-k",
111
  minimum=1,
112
  maximum=1000,
113
  step=1,
114
+ value=3,
115
  ),
116
  gr.Slider(
117
  label="Repetition penalty",