Spaces:
Sleeping
Sleeping
[update]add main
Browse files
main.py
CHANGED
@@ -98,13 +98,13 @@ def main():
|
|
98 |
|
99 |
with gr.Row():
|
100 |
with gr.Column(scale=1):
|
101 |
-
max_new_tokens = gr.Slider(minimum=0, maximum=512, value=512, step=1, label="max_new_tokens")
|
102 |
with gr.Column(scale=1):
|
103 |
-
top_p = gr.Slider(minimum=0, maximum=1, value=0.85, step=0.01, label="top_p")
|
104 |
with gr.Column(scale=1):
|
105 |
-
temperature = gr.Slider(minimum=0, maximum=1, value=0.35, step=0.01, label="temperature")
|
106 |
with gr.Column(scale=1):
|
107 |
-
repetition_penalty = gr.Slider(minimum=0, maximum=2, value=1.2, step=0.01, label="repetition_penalty")
|
108 |
|
109 |
with gr.Row():
|
110 |
model_name = gr.Dropdown(choices=["Qwen/Qwen-7B-Chat"],
|
|
|
98 |
|
99 |
with gr.Row():
|
100 |
with gr.Column(scale=1):
|
101 |
+
max_new_tokens = gr.Slider(minimum=0, maximum=512, value=512, step=1, label="max_new_tokens")
|
102 |
with gr.Column(scale=1):
|
103 |
+
top_p = gr.Slider(minimum=0, maximum=1, value=0.85, step=0.01, label="top_p")
|
104 |
with gr.Column(scale=1):
|
105 |
+
temperature = gr.Slider(minimum=0, maximum=1, value=0.35, step=0.01, label="temperature")
|
106 |
with gr.Column(scale=1):
|
107 |
+
repetition_penalty = gr.Slider(minimum=0, maximum=2, value=1.2, step=0.01, label="repetition_penalty")
|
108 |
|
109 |
with gr.Row():
|
110 |
model_name = gr.Dropdown(choices=["Qwen/Qwen-7B-Chat"],
|