update readme
This commit is contained in:
@@ -46,43 +46,30 @@ un-embedding. On top of it, J-Wash adds:
|
|||||||
Pre-fitted lenses come from [Neuronpedia](https://huggingface.co/neuronpedia/jacobian-lens);
|
Pre-fitted lenses come from [Neuronpedia](https://huggingface.co/neuronpedia/jacobian-lens);
|
||||||
you can also fit your own locally.
|
you can also fit your own locally.
|
||||||
|
|
||||||
## Requirements
|
|
||||||
|
|
||||||
- An **NVIDIA GPU (CUDA)**. J-Wash is CUDA-only (it uses `pynvml` and `cuda:*`
|
|
||||||
devices); there is no CPU/MPS path. ~12 GB of VRAM comfortably runs a 4B model.
|
|
||||||
- **Python 3.13** (3.11+ should work).
|
|
||||||
- **Node.js 18+** and npm (to build the front-end).
|
|
||||||
|
|
||||||
## Installation
|
## Installation
|
||||||
|
|
||||||
|
Requirements:
|
||||||
|
- NVIDIA GPU (CUDA)
|
||||||
|
- Python 3.11+
|
||||||
|
- Node.js 18+
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
# 1. Clone this repo
|
git clone https://github.com/extraltodeus/j-wash.git
|
||||||
git clone https://github.com/<you>/j-wash.git
|
|
||||||
cd j-wash
|
cd j-wash
|
||||||
|
|
||||||
# 2. Create an environment (conda or venv)
|
|
||||||
conda create -n jwash python=3.13 -y
|
|
||||||
conda activate jwash
|
|
||||||
# or: python -m venv .venv && source .venv/bin/activate (Windows: .venv\Scripts\activate)
|
|
||||||
|
|
||||||
# 3. Install PyTorch matching your CUDA version — see https://pytorch.org
|
|
||||||
# (Blackwell / RTX 50-series needs a cu128+ build)
|
|
||||||
pip install torch --index-url https://download.pytorch.org/whl/cu124
|
pip install torch --index-url https://download.pytorch.org/whl/cu124
|
||||||
|
|
||||||
# 4. Vendor the Jacobian-lens library (not on PyPI) and install it editable
|
|
||||||
git clone https://github.com/anthropics/jacobian-lens vendor/jacobian-lens
|
git clone https://github.com/anthropics/jacobian-lens vendor/jacobian-lens
|
||||||
pip install -e vendor/jacobian-lens
|
|
||||||
|
|
||||||
# 5. Install the rest
|
pip install -e vendor/jacobian-lens
|
||||||
pip install -r requirements.txt
|
pip install -r requirements.txt
|
||||||
|
|
||||||
# 6. Build the front-end
|
|
||||||
cd ui && npm install && npm run build && cd ..
|
cd ui && npm install && npm run build && cd ..
|
||||||
|
|
||||||
|
python -X utf8 run.py
|
||||||
```
|
```
|
||||||
|
|
||||||
> The `jlens` package is imported as `jlens` in the code. If Anthropic's repo URL
|
Then open **http://localhost:8381**.
|
||||||
> changes, update step 4 accordingly — any package exposing `jlens.lens.JacobianLens`,
|
|
||||||
> `jlens.from_hf`, `jlens.fit`, `jlens.vis` and `jlens.examples` will work.
|
|
||||||
|
|
||||||
## Running
|
## Running
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user