Update app.py
Browse files
app.py
CHANGED
@@ -404,7 +404,7 @@ with gr.Blocks() as demo:
|
|
404 |
)
|
405 |
|
406 |
calc_flops_button = gr.Button("Calculate FLOPs")
|
407 |
-
flops_result = gr.JSON(label="FLOP Calculation Result"
|
408 |
calc_flops_button.click(
|
409 |
calc_flops,
|
410 |
inputs=[vocab_size, hidden_size, sequence_length, num_layers, kv_size_ratio, topk, moe, num_experts, expert_interval, batch_size, tokens, checkpoint_activations, ffn_expansion_factor, infer],
|
|
|
404 |
)
|
405 |
|
406 |
calc_flops_button = gr.Button("Calculate FLOPs")
|
407 |
+
flops_result = gr.JSON(label="FLOP Calculation Result")
|
408 |
calc_flops_button.click(
|
409 |
calc_flops,
|
410 |
inputs=[vocab_size, hidden_size, sequence_length, num_layers, kv_size_ratio, topk, moe, num_experts, expert_interval, batch_size, tokens, checkpoint_activations, ffn_expansion_factor, infer],
|