mirror of
https://github.com/wassname/weight-steering.git
synced 2026-06-27 15:18:02 +08:00
wassname
40 lines
886 B
TOML
40 lines
886 B
TOML
[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"
|