tidy

2026-06-27 16:48:01 +08:00 · 2026-04-25 19:27:53 +08:00
parent 1c0152910a
commit 4ad6971038
53 changed files with 950 additions and 4845 deletions
@@ -0,0 +1,39 @@
+[project]
+name = "weight-steering"
+version = "0.1.0"
+description = "Fork of Anthropic's weight-steering: replicate, test SVD/subspace alignment, sweep PEFT adapter families."
+requires-python = ">=3.11"
+dependencies = [
+    "torch>=2.4",
+    "transformers>=4.46",
+    "peft>=0.13",                # DeLoRA, DoRA, init_lora_weights="pissa"
+    "datasets>=3.0",
+    "accelerate>=1.0",
+    "einops>=0.8",
+    "jaxtyping>=0.2",
+    "beartype>=0.19",
+    "loguru>=0.7",
+    "polars>=1.10",
+    "tabulate>=0.9",
+    "wandb>=0.18",
+    "tyro>=0.8",
+    "baukit @ git+https://github.com/davidbau/baukit.git",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=8",
+    "ipykernel>=6",
+    "ruff>=0.7",
+]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/ws"]
+
+[tool.ruff]
+line-length = 100
+target-version = "py311"