Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -50,8 +50,8 @@ model = OlmoeForCausalLM.from_pretrained(
|
|
50 |
def stream_chat(
|
51 |
message: str,
|
52 |
history: list,
|
53 |
-
temperature: float = 0.
|
54 |
-
max_new_tokens: int =
|
55 |
top_p: float = 1.0,
|
56 |
top_k: int = 20,
|
57 |
penalty: float = 1.2,
|
@@ -110,15 +110,15 @@ with gr.Blocks(css=CSS, theme="Nymbo/Nymbo_Theme") as demo:
|
|
110 |
minimum=0,
|
111 |
maximum=1,
|
112 |
step=0.1,
|
113 |
-
value=0.
|
114 |
label="Temperature",
|
115 |
render=False,
|
116 |
),
|
117 |
gr.Slider(
|
118 |
minimum=128,
|
119 |
-
maximum=
|
120 |
step=1,
|
121 |
-
value=
|
122 |
label="Max new tokens",
|
123 |
render=False,
|
124 |
),
|
|
|
50 |
def stream_chat(
|
51 |
message: str,
|
52 |
history: list,
|
53 |
+
temperature: float = 0.5,
|
54 |
+
max_new_tokens: int = 256,
|
55 |
top_p: float = 1.0,
|
56 |
top_k: int = 20,
|
57 |
penalty: float = 1.2,
|
|
|
110 |
minimum=0,
|
111 |
maximum=1,
|
112 |
step=0.1,
|
113 |
+
value=0.5,
|
114 |
label="Temperature",
|
115 |
render=False,
|
116 |
),
|
117 |
gr.Slider(
|
118 |
minimum=128,
|
119 |
+
maximum=4096,
|
120 |
step=1,
|
121 |
+
value=256,
|
122 |
label="Max new tokens",
|
123 |
render=False,
|
124 |
),
|