aksell commited on
Commit
cea9292
·
1 Parent(s): 2e81150

Reword checkbox to hide attention to tokens

Browse files
hexviz/pages/1_🗺️Identify_Interesting_Heads.py CHANGED
@@ -82,7 +82,9 @@ sequence = res_to_1letter(residues)
82
  l = len(sequence)
83
  slice_start, slice_end = select_sequence_slice(l)
84
  truncated_sequence = sequence[slice_start - 1 : slice_end]
85
- remove_special_tokens = st.sidebar.checkbox("Remove special tokens", key="remove_special_tokens")
 
 
86
 
87
 
88
  layer_sequence, head_sequence = select_heads_and_layers(st.sidebar, selected_model)
 
82
  l = len(sequence)
83
  slice_start, slice_end = select_sequence_slice(l)
84
  truncated_sequence = sequence[slice_start - 1 : slice_end]
85
+ remove_special_tokens = st.sidebar.checkbox(
86
+ "Hide attention to special tokens", key="remove_special_tokens"
87
+ )
88
 
89
 
90
  layer_sequence, head_sequence = select_heads_and_layers(st.sidebar, selected_model)