Spaces:
Running
Running
Commit ·
5357354
1
Parent(s): 4fab954
fix manual
Browse files
app.py
CHANGED
|
@@ -128,11 +128,22 @@ with gr.Blocks() as demo:
|
|
| 128 |
|
| 129 |
# Handle manual input
|
| 130 |
def manual_input_to_config(*args):
|
| 131 |
-
config =
|
| 132 |
-
'hidden_size'
|
| 133 |
-
'
|
| 134 |
-
'
|
| 135 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 136 |
return process_yaml_and_plot(config)
|
| 137 |
|
| 138 |
manual_submit.click(
|
|
|
|
| 128 |
|
| 129 |
# Handle manual input
|
| 130 |
def manual_input_to_config(*args):
|
| 131 |
+
config = {
|
| 132 |
+
'hidden_size': args[0],
|
| 133 |
+
'num_layers': args[3],
|
| 134 |
+
'vocab_size': args[4],
|
| 135 |
+
'intermediate_size': args[5],
|
| 136 |
+
'seq_len': args[6],
|
| 137 |
+
'mbs': args[7],
|
| 138 |
+
'batch_accum': args[8],
|
| 139 |
+
'tp': args[9],
|
| 140 |
+
'pp': args[10],
|
| 141 |
+
'dp': args[11],
|
| 142 |
+
'zero_stage': args[12],
|
| 143 |
+
'tie_word_embeddings': args[13],
|
| 144 |
+
'num_attention_heads': args[1],
|
| 145 |
+
'num_key_value_heads': args[2]
|
| 146 |
+
}
|
| 147 |
return process_yaml_and_plot(config)
|
| 148 |
|
| 149 |
manual_submit.click(
|