fix: load probe class with weights_only = False

Author: adamkarvonen

sae_bench/evals/absorption/common.py

@@ -73,6 +73,7 @@ def load_probe(
     probe = torch.load(
         Path(probes_dir) / f"{model_name}" / f"layer_{layer}" / "probe.pth",
         map_location=device,
+        weights_only=False,
     ).to(dtype=dtype)
     return probe