idk, probs good

jbloomAus · Jan 27, 2024 · 0407ad9 · 0407ad9
1 parent 1ec8f97
commit 0407ad9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sae_analysis/visualizer/data_fns.py b/sae_analysis/visualizer/data_fns.py
@@ -820,7 +820,8 @@ def hook_fn_resid_post(resid_post: Float[Tensor, "batch seq d_model"], hook: Hoo
         _logits = logits[feat]
 
         # Get data for logits histogram
-        logits_histogram_data.append(HistogramData(_logits, n_bins=40, tickmode="5 ticks"))
+        # logits_histogram_data.append(HistogramData(_logits, n_bins=40, tickmode="5 ticks"))
+        logits_histogram_data.append(HistogramData(_logits, n_bins=40, tickmode="ints"))
 
         # Get data for logits table
         top10_logits.append((TopK(_logits.topk(k=10, largest=False)), TopK(_logits.topk(k=10))))