Spaces:

taka-yamakoshi
/

causal-intervention-demo

Runtime error

taka-yamakoshi commited on Mar 26, 2023

Commit

842f713

1 Parent(s): a267a6b

debug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -248,7 +248,6 @@ if __name__=='__main__':
             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)
-            print(probs_original)
     if st.session_state['page_status'] == 'finish_debug':
             for layer_id in range(num_layers):

             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):