Peter commited on
Commit
146762d
1 Parent(s): 6b6c220

adjust defaults

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -217,7 +217,7 @@ if __name__ == "__main__":
217
  minimum=0.0, maximum=1.0, step=0.01, default=0.6, label="temperature"
218
  ),
219
  Slider(minimum=0.0, maximum=1.0, step=0.01, default=0.95, label="top_p"),
220
- Slider(minimum=0, maximum=250, step=5, default=50, label="top_k"),
221
  ],
222
  outputs="html",
223
  examples_per_page=8,
@@ -228,10 +228,10 @@ if __name__ == "__main__":
228
  ["Happy birthday!", 0.9, 0.95, 50],
229
  ["I have a question, can you help me?", 0.6, 0.95, 50],
230
  ["Do you know a joke?", 0.8, 0.85, 50],
231
- ["Will you marry me?", 0.9, 0.95, 138],
232
- ["Are you single?", 0.6, 0.95, 138],
233
  ["Do you like people?", 0.7, 0.95, 25],
234
- ["You never took a short cut before?", 0.7, 0.95, 125],
235
  ],
236
  title=f"GPT Chatbot Demo: {default_model} Model",
237
  description=f"A Demo of a Chatbot trained for conversation with humans. Size XL= 1.5B parameters.\n\n"
 
217
  minimum=0.0, maximum=1.0, step=0.01, default=0.6, label="temperature"
218
  ),
219
  Slider(minimum=0.0, maximum=1.0, step=0.01, default=0.95, label="top_p"),
220
+ Slider(minimum=0, maximum=100, step=5, default=20, label="top_k"),
221
  ],
222
  outputs="html",
223
  examples_per_page=8,
 
228
  ["Happy birthday!", 0.9, 0.95, 50],
229
  ["I have a question, can you help me?", 0.6, 0.95, 50],
230
  ["Do you know a joke?", 0.8, 0.85, 50],
231
+ ["Will you marry me?", 0.9, 0.95, 100],
232
+ ["Are you single?", 0.6, 0.95, 100],
233
  ["Do you like people?", 0.7, 0.95, 25],
234
+ ["You never took a short cut before?", 0.7, 0.95, 100],
235
  ],
236
  title=f"GPT Chatbot Demo: {default_model} Model",
237
  description=f"A Demo of a Chatbot trained for conversation with humans. Size XL= 1.5B parameters.\n\n"