nouamanetazi HF Staff commited on
Commit
5357354
·
1 Parent(s): 4fab954

fix manual

Browse files
Files changed (1) hide show
  1. app.py +16 -5
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 = dict(zip([
132
- 'hidden_size', 'num_layers', 'vocab_size', 'intermediate_size',
133
- 'seq_len', 'mbs', 'batch_accum', 'tp', 'pp', 'dp', 'zero_stage',
134
- 'tie_word_embeddings', 'num_attention_heads', 'num_key_value_heads'
135
- ], args))
 
 
 
 
 
 
 
 
 
 
 
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(