Spaces:

ma7583
/

flops_calc

Sleeping

ma7583 commited on May 19

Commit

b4db2f5

verified ·

1 Parent(s): 4191ca6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -51,7 +51,7 @@ def estimate_transformer_stats(batch_size, seq_len, num_layers, hidden_dim, voca
 iface = gr.Interface(
     fn=estimate_transformer_stats,
     inputs=[
-        gr.Number(label="Batch Size", value=32),
         gr.Number(label="Sequence Length", value=2048),
         gr.Number(label="Number of Layers", value=24),
         gr.Number(label="Hidden Size (d_model)", value=2048),

 iface = gr.Interface(
     fn=estimate_transformer_stats,
     inputs=[
+        gr.Number(label="Batch Size", value=1),
         gr.Number(label="Sequence Length", value=2048),
         gr.Number(label="Number of Layers", value=24),
         gr.Number(label="Hidden Size (d_model)", value=2048),