Spaces:
Sleeping
Sleeping
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(
|
|
|
|
|
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)
|