Hugging Face's logo

Spaces:
AI4PD
/
hexviz
Sleeping

App Files Community
1
aksell commited on
Commit
b5726b3
·
1 Parent(s): 491bdcd

Remove debug print

Browse files
Files changed (1) hide show
  1. hexviz/pages/1_🗺️Identify_Interesting_Heads.py +0 -1
hexviz/pages/1_🗺️Identify_Interesting_Heads.py CHANGED
@@ -75,7 +75,6 @@ attention = get_attention(
75
  model_type=selected_model.name,
76
  remove_special_tokens=True,
77
  )
78
- st.write(attention.shape)
79
 
80
  fig = plot_tiled_heatmap(
81
  attention, layer_sequence=layer_sequence, head_sequence=head_sequence
 
75
  model_type=selected_model.name,
76
  remove_special_tokens=True,
77
  )
 
78
 
79
  fig = plot_tiled_heatmap(
80
  attention, layer_sequence=layer_sequence, head_sequence=head_sequence