stevengrove commited on
Commit
551cd0e
1 Parent(s): 7ff898d

revise default setting

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -46,7 +46,7 @@ if __name__ == '__main__':
46
  parser.add_argument('--top_p', type=float, default=1.0, help='top_p for the llm model')
47
  parser.add_argument('--top_k', type=int, default=40, help='top_k for the llm model')
48
  parser.add_argument('--num_beams', type=int, default=4, help='num_beams for the llm model')
49
- parser.add_argument('--keep_last_n_paragraphs', type=int, default=1, help='keep last n paragraphs in the memory')
50
  args = parser.parse_args()
51
 
52
  examples = [
@@ -81,10 +81,10 @@ if __name__ == '__main__':
81
  interactive=True,
82
  label="Remember Last N Paragraphs")
83
  max_new_token = gr.Slider(
84
- minimum=128,
85
- maximum=1024,
86
  value=args.max_new_tokens,
87
- step=64,
88
  interactive=True,
89
  label="Max New Tokens")
90
  temperature = gr.Slider(
 
46
  parser.add_argument('--top_p', type=float, default=1.0, help='top_p for the llm model')
47
  parser.add_argument('--top_k', type=int, default=40, help='top_k for the llm model')
48
  parser.add_argument('--num_beams', type=int, default=4, help='num_beams for the llm model')
49
+ parser.add_argument('--keep_last_n_paragraphs', type=int, default=0, help='keep last n paragraphs in the memory')
50
  args = parser.parse_args()
51
 
52
  examples = [
 
81
  interactive=True,
82
  label="Remember Last N Paragraphs")
83
  max_new_token = gr.Slider(
84
+ minimum=64,
85
+ maximum=512,
86
  value=args.max_new_tokens,
87
+ step=1,
88
  interactive=True,
89
  label="Max New Tokens")
90
  temperature = gr.Slider(