taka-yamakoshi
commited on
Commit
•
271977c
1
Parent(s):
c825935
simplify setup
Browse files- app.py +1 -0
- packages.txt +0 -1
- requirements.txt +0 -1
app.py
CHANGED
|
@@ -253,3 +253,4 @@ if __name__=='__main__':
|
|
| 253 |
for layer_id in range(num_layers):
|
| 254 |
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)]
|
| 255 |
probs = run_intervention(interventions,num_heads,model,masked_ids_option_1,masked_ids_option_2,option_1_tokens,option_2_tokens,pron_locs)
|
|
|
|
|
|
| 253 |
for layer_id in range(num_layers):
|
| 254 |
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)]
|
| 255 |
probs = run_intervention(interventions,num_heads,model,masked_ids_option_1,masked_ids_option_2,option_1_tokens,option_2_tokens,pron_locs)
|
| 256 |
+
|
packages.txt
CHANGED
|
@@ -1 +0,0 @@
|
|
| 1 |
-
libgl1-mesa-dev
|
|
|
|
|
|
requirements.txt
CHANGED
|
@@ -1,5 +1,4 @@
|
|
| 1 |
torch
|
| 2 |
transformers
|
| 3 |
seaborn
|
| 4 |
-
protobuf~=3.19.0
|
| 5 |
sentencepiece
|
|
|
|
| 1 |
torch
|
| 2 |
transformers
|
| 3 |
seaborn
|
|
|
|
| 4 |
sentencepiece
|