taka-yamakoshi commited on
Commit
5b8028c
1 Parent(s): e919fae
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -162,10 +162,10 @@ def run_intervention(interventions,batch_size,model,masked_ids_option_1,masked_i
162
  if __name__=='__main__':
163
  wide_setup()
164
  #load_css('style.css')
165
- tokenizer,model = load_model()
166
- num_layers, num_heads = model.config.num_hidden_layers, model.config.num_attention_heads
167
- st.write(num_layers,num_heads)
168
- mask_id = tokenizer('[MASK]').input_ids[1:-1][0]
169
 
170
  main_area = st.empty()
171
 
 
162
  if __name__=='__main__':
163
  wide_setup()
164
  #load_css('style.css')
165
+ #tokenizer,model = load_model()
166
+ #num_layers, num_heads = model.config.num_hidden_layers, model.config.num_attention_heads
167
+ #st.write(num_layers,num_heads)
168
+ #mask_id = tokenizer('[MASK]').input_ids[1:-1][0]
169
 
170
  main_area = st.empty()
171