qgyd2021 commited on
Commit
4e39876
1 Parent(s): 4f49487

[update]add main

Browse files
Files changed (1) hide show
  1. main.py +4 -4
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"],