TING2938 commited on
Commit
23a3f8b
·
1 Parent(s): 6ff1d7a
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -27,8 +27,12 @@ with gr.Blocks() as demo:
27
  with gr.Tab("calc_extra_token_length"):
28
  with gr.Row():
29
  with gr.Column():
30
- inputs = [gr.Number(4096), gr.Number(
31
- 10000), gr.Number(1000000), gr.Number(128)]
 
 
 
 
32
  btn = gr.Button("calc")
33
  outputs = gr.Textbox(value=0, label="extra max token (k)")
34
  btn.click(fn=calc_extra_token_length, inputs=inputs, outputs=outputs)
 
27
  with gr.Tab("calc_extra_token_length"):
28
  with gr.Row():
29
  with gr.Column():
30
+ inputs = [
31
+ gr.Number(4096, label="old_seq_len"),
32
+ gr.Number(10000, label="old_base"),
33
+ gr.Number(1000000, label="new_base"),
34
+ gr.Number(128, label="dim"),
35
+ ]
36
  btn = gr.Button("calc")
37
  outputs = gr.Textbox(value=0, label="extra max token (k)")
38
  btn.click(fn=calc_extra_token_length, inputs=inputs, outputs=outputs)