Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -71,9 +71,9 @@ with gr.Blocks() as demo:
|
|
71 |
max_new_tokens = gr.Slider(
|
72 |
label='Max new tokens',
|
73 |
minimum=1,
|
74 |
-
maximum=
|
75 |
step=1,
|
76 |
-
value=
|
77 |
)
|
78 |
temperature = gr.Slider(
|
79 |
label='Temperature',
|
@@ -118,7 +118,7 @@ with gr.Blocks() as demo:
|
|
118 |
"model": MODEL_NAME,
|
119 |
"prompt": str(message),
|
120 |
"temperature": float(temperature),
|
121 |
-
"n":
|
122 |
"max_tokens": int(max_new_tokens),
|
123 |
"stop": "",
|
124 |
"top_p": float(top_p),
|
|
|
71 |
max_new_tokens = gr.Slider(
|
72 |
label='Max new tokens',
|
73 |
minimum=1,
|
74 |
+
maximum=1024,
|
75 |
step=1,
|
76 |
+
value=512,
|
77 |
)
|
78 |
temperature = gr.Slider(
|
79 |
label='Temperature',
|
|
|
118 |
"model": MODEL_NAME,
|
119 |
"prompt": str(message),
|
120 |
"temperature": float(temperature),
|
121 |
+
"n": int(top_k),
|
122 |
"max_tokens": int(max_new_tokens),
|
123 |
"stop": "",
|
124 |
"top_p": float(top_p),
|