Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ def stream_chat(
|
|
47 |
temperature: float = 0.8,
|
48 |
max_new_tokens: int = 1024,
|
49 |
top_p: float = 1.0,
|
50 |
-
top_k: int =
|
51 |
penalty: float = 1.2,
|
52 |
):
|
53 |
print(f'message: {message}')
|
@@ -132,9 +132,9 @@ with gr.Blocks(css=CSS, theme="soft") as demo:
|
|
132 |
),
|
133 |
gr.Slider(
|
134 |
minimum=1,
|
135 |
-
maximum=
|
136 |
step=1,
|
137 |
-
value=
|
138 |
label="top_k",
|
139 |
render=False,
|
140 |
),
|
|
|
47 |
temperature: float = 0.8,
|
48 |
max_new_tokens: int = 1024,
|
49 |
top_p: float = 1.0,
|
50 |
+
top_k: int = 50,
|
51 |
penalty: float = 1.2,
|
52 |
):
|
53 |
print(f'message: {message}')
|
|
|
132 |
),
|
133 |
gr.Slider(
|
134 |
minimum=1,
|
135 |
+
maximum=50,
|
136 |
step=1,
|
137 |
+
value=50,
|
138 |
label="top_k",
|
139 |
render=False,
|
140 |
),
|