Don't skip heads and layers by deafult

hexviz/Attention_Visualization.py

@@ -58,7 +58,6 @@ label_highest = st.sidebar.checkbox("Label highest attention pairs", value=True)
 
 
 
-
 left, mid, right = st.columns(3)
 with left:
     selected_model_name = st.selectbox("Select a model", [model.name.value for model in models], index=0)
@@ -70,7 +69,6 @@ with right:
     head_one = st.number_input("Head", value=1, min_value=1, max_value=selected_model.heads)
     head = head_one - 1
 
-
 if selected_model.name == ModelType.ZymCTRL:
     try:
         ec_class = structure.header["compound"]["1"]["ec"]

hexviz/pages/Identify_interesting_Heads.py

@@ -41,9 +41,9 @@ slice_start, slice_end = st.sidebar.slider("Sequence", min_value=1, max_value=l,
 # slice_end = st.sidebar.number_input(f"Section end(1-{l})",value=50, min_value=1, max_value=l)
 truncated_sequence = sequence[slice_start-1:slice_end]
 
-head_range = st.sidebar.slider("Heads to plot", min_value=1, max_value=selected_model.heads, value=(1, selected_model.heads), step=1)
-layer_range = st.sidebar.slider("Layers to plot", min_value=1, max_value=selected_model.layers, value=(1, selected_model.layers), step=1)
-step_size = st.sidebar.number_input("Optional step size to skip heads and layers", value=2, min_value=1, max_value=selected_model.layers)
+head_range = st.sidebar.slider("Heads to plot", min_value=1, max_value=selected_model.heads, value=(1, selected_model.heads//2), step=1)
+layer_range = st.sidebar.slider("Layers to plot", min_value=1, max_value=selected_model.layers, value=(1, selected_model.layers//2), step=1)
+step_size = st.sidebar.number_input("Optional step size to skip heads and layers", value=1, min_value=1, max_value=selected_model.layers)
 layer_sequence = list(range(layer_range[0]-1, layer_range[1], step_size))
 head_sequence = list(range(head_range[0]-1, head_range[1], step_size))