taka-yamakoshi
commited on
Commit
•
842f713
1
Parent(s):
a267a6b
debug
Browse files
app.py
CHANGED
|
@@ -248,7 +248,6 @@ if __name__=='__main__':
|
|
| 248 |
interventions = [{'lay':[],'qry':[],'key':[],'val':[]} for i in range(num_layers)]
|
| 249 |
probs_original = run(interventions,1,model,masked_ids_option_1,masked_ids_option_2,option_1_tokens,option_2_tokens,pron_locs)
|
| 250 |
st.write(probs_original)
|
| 251 |
-
print(probs_original)
|
| 252 |
|
| 253 |
if st.session_state['page_status'] == 'finish_debug':
|
| 254 |
for layer_id in range(num_layers):
|
|
|
|
| 248 |
interventions = [{'lay':[],'qry':[],'key':[],'val':[]} for i in range(num_layers)]
|
| 249 |
probs_original = run(interventions,1,model,masked_ids_option_1,masked_ids_option_2,option_1_tokens,option_2_tokens,pron_locs)
|
| 250 |
st.write(probs_original)
|
|
|
|
| 251 |
|
| 252 |
if st.session_state['page_status'] == 'finish_debug':
|
| 253 |
for layer_id in range(num_layers):
|