Bleeding Edge - Accept All Pull Requests#22

Open

Tauheed-Elahee wants to merge 32 commits into

JoshEngels:mainfrom

Tauheed-Elahee:bleeding-edge

Commits on Mar 13, 2026

Fix some typos in README.md

JasonGross
authored and
Tauheed-Elahee
committed
README: add some hyperlinks and code font

JasonGross
authored and
Tauheed-Elahee
committed
`legendHandles` => `legend_handles`

JasonGross
authored and
Tauheed-Elahee
committed
Fix to support MPS: convert to float32 earlier

JasonGross
authored and
Tauheed-Elahee
committed
Don't fail with "AssertionError: Not enough CUDA devices to support n_devices 2" when there aren't 2 GPUs

JasonGross
authored and
Tauheed-Elahee
committed
Revert to cpu if cuda is not available

JasonGross
authored and
Tauheed-Elahee
committed
Use pathlib more and have default cache paths be relative to the repo directory

JasonGross
authored and
Tauheed-Elahee
committed
Add support for loading 16-bit models

JasonGross
authored and
Tauheed-Elahee
committed
Add use of dtype in ipynb

JasonGross
authored and
Tauheed-Elahee
committed
Update torch.load calls to include weights_only parameter
Tauheed-Elahee
committed
Update all calls to torch.load with required weights_only parameter.
Tauheed-Elahee
committed
Fix TransformerLens model name for GPT-2
Tauheed-Elahee
committed
Use dynamic device count instead of hardcoded num_devices=2
Tauheed-Elahee
committed
Add compatibility for newer sae_lens forward() return type
Tauheed-Elahee
committed
Fix incorrect --clustering_type arg in README examples
Tauheed-Elahee
committed
Merge remote-tracking branch 'JasonGross/patch-2' into bleeding-edge
Tauheed-Elahee
committed
Merge remote-tracking branch 'JasonGross/patch-3' into bleeding-edge
Tauheed-Elahee
committed
Merge remote-tracking branch 'JasonGross/legend_handles' into bleeding-edge
Tauheed-Elahee
committed
Merge remote-tracking branch 'JasonGross/mps-cleanup' into bleeding-edge
Tauheed-Elahee
committed
Merge remote-tracking branch 'JasonGross/patch-1' into bleeding-edge
Tauheed-Elahee
committed
Merge remote-tracking branch 'JasonGross/cuda-available-simpler' into bleeding-edge
Tauheed-Elahee
committed
Merge remote-tracking branch 'JasonGross/pathlib' into bleeding-edge
Tauheed-Elahee
committed
Merge PR #10 (dtype support) with conflicts resolved: keep PR #11 device handling + add dtype
Tauheed-Elahee
committed
Merge PR #16 (torch.load weights_only) with conflicts resolved: keep pathlib paths + add weights_only=False
Tauheed-Elahee
committed
Merge branch 'tl-model-name' into bleeding-edge
Tauheed-Elahee
committed
Merge branch 'dynamic-num-devices' into bleeding-edge
Tauheed-Elahee
committed
Merge branch 'sae-lens-compatibility' into bleeding-edge
Tauheed-Elahee
committed
Merge branch 'readme-method-arg' into bleeding-edge
Tauheed-Elahee
committed
Read BASE_DIR from environment variable with fallback to default
Tauheed-Elahee
committed
Change default CUDA device from cuda:4 to cuda:0
Tauheed-Elahee
committed
Remove NVIDIA CUDA 12.1 hard pins from requirements
Tauheed-Elahee
committed
Add requirements.in and update requirements.txt via pip-compile
Tauheed-Elahee
committed