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(
|