Spaces:
Runtime error
Runtime error
stevengrove
commited on
Commit
•
551cd0e
1
Parent(s):
7ff898d
revise default setting
Browse files
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=
|
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=
|
85 |
-
maximum=
|
86 |
value=args.max_new_tokens,
|
87 |
-
step=
|
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(
|