debug

app.py

@@ -245,11 +245,11 @@ if __name__=='__main__':
             option_1_tokens = option_1_tokens_1
             option_2_tokens = option_2_tokens_1
 
+    if st.session_state['page_status'] == 'finish_debug':
             interventions = [{'lay':[],'qry':[],'key':[],'val':[]} for i in range(num_layers)]
             probs_original = run(interventions,1,model,masked_ids_option_1,masked_ids_option_2,option_1_tokens,option_2_tokens,pron_locs)
             st.write(probs_original)
 
-    if st.session_state['page_status'] == 'finish_debug':
             for layer_id in range(num_layers):
                 interventions = [create_interventions(16,['lay','qry','key','val'],num_heads) if i==layer_id else {'lay':[],'qry':[],'key':[],'val':[]} for i in range(num_layers)]
                 probs = run(interventions,num_heads,model,masked_ids_option_1,masked_ids_option_2,option_1_tokens,option_2_tokens,pron_locs)