adamcasson
commited on
Commit
·
6ecfb5c
1
Parent(s):
64e5a86
add markdown title
Browse files
app.py
CHANGED
@@ -42,6 +42,10 @@ def calculator(n_layer, d_model, n_heads, n_vocab, n_ctx, ff_ratio, incl_embed):
|
|
42 |
|
43 |
|
44 |
with gr.Blocks() as iface:
|
|
|
|
|
|
|
|
|
45 |
with gr.Row():
|
46 |
with gr.Column():
|
47 |
n_layer = gr.Number(label="Number of layers (n_layer)")
|
|
|
42 |
|
43 |
|
44 |
with gr.Blocks() as iface:
|
45 |
+
gr.Markdown("## Transformer FLOPs Calculator")
|
46 |
+
gr.Markdown(
|
47 |
+
"Calculate how many FLOPs a Transformer language model has using the method described in [DeepMind's Chinchilla scaling law paper](https://arxiv.org/abs/2203.15556) (see Appendix F)."
|
48 |
+
)
|
49 |
with gr.Row():
|
50 |
with gr.Column():
|
51 |
n_layer = gr.Number(label="Number of layers (n_layer)")
|