taka-yamakoshi
commited on
Commit
•
5b8028c
1
Parent(s):
e919fae
debug
Browse files
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 |
|