mirror of
https://github.com/wassname/AntiPaSTO.git
synced 2026-08-20 12:00:30 +08:00
82 lines
2.0 KiB
TOML
82 lines
2.0 KiB
TOML
[project]
|
|
name = "antipasto"
|
|
version = "0.5.0"
|
|
description = "Self-supervised honesty steering via anti-parallel representations"
|
|
authors = [{ name = "Michael J Clark" }]
|
|
repository = "https://github.com/wassname/AntiPaSTO"
|
|
license = { file = "LICENSE" }
|
|
readme = "README.md"
|
|
requires-python = ">=3.10"
|
|
dependencies = [
|
|
# Core ML
|
|
"numpy>=1.26.4",
|
|
"torch>=2.1.2",
|
|
"transformers[torch]>4.51.0",
|
|
"peft",
|
|
"bitsandbytes>=0.47.0",
|
|
"einops>=0.8.1",
|
|
"jaxtyping>=0.3.2",
|
|
# Data/utils
|
|
"datasets>=4.1.1",
|
|
"pandas>=2.3.2",
|
|
"tqdm>=4.66.1",
|
|
"loguru>=0.7.3",
|
|
# Model internals
|
|
"baukit",
|
|
"scikit-learn>=1.4.0",
|
|
"scipy>=1.11.0",
|
|
# Export
|
|
"gguf>=0.13.0",
|
|
"safetensors>=0.4.0",
|
|
"cattrs>=25.3.0",
|
|
"tyro>=1.0.5",
|
|
"tabulate>=0.9.0",
|
|
"wandb>=0.23.0",
|
|
]
|
|
|
|
[dependency-groups]
|
|
dev = [
|
|
"pytest>=8.0.2",
|
|
"ruff>=0.8.3",
|
|
"ipykernel>=6.30.1",
|
|
"ipywidgets>=8.1.7",
|
|
]
|
|
nbs = [
|
|
# Visualization
|
|
"matplotlib>=3.10.6",
|
|
"seaborn>=0.13.2",
|
|
"plotly>=6.5.0",
|
|
"great-tables>=0.20.0",
|
|
"adjusttext>=1.3.0",
|
|
# Experiment tracking
|
|
"wandb>=0.22.3",
|
|
# Notebook utils
|
|
"ipykernel>=6.30.1",
|
|
"ipywidgets>=8.1.7",
|
|
"jupytext>=1.18.1",
|
|
"nbformat>=5.10.4",
|
|
"tabulate>=0.9.0",
|
|
# Baselines
|
|
"repeng>=0.4.0",
|
|
# Analysis
|
|
"umap-learn>=0.5.9.post2",
|
|
]
|
|
|
|
[tool.ruff.lint]
|
|
# F722: syntax error in forward annotation - triggered by jaxtyping's string shape annotations
|
|
# See: https://docs.kidger.site/jaxtyping/faq/
|
|
ignore = ["F722"]
|
|
|
|
[tool.pytest.ini_options]
|
|
# python_files = ["tests.py"]
|
|
markers = ["slow: marks tests as slow (deselect with '-m \"not slow\"')"]
|
|
|
|
[tool.uv.sources]
|
|
baukit = { git = "https://github.com/davidbau/baukit.git" }
|
|
peft = { git = "https://github.com/huggingface/peft.git", rev = "41091ecd31abc84bda9e899c4e80ec96f3fa99b2" }
|
|
repeng = { git = "https://github.com/wassname/repeng.git", branch="my_fixes" }
|
|
|
|
[build-system]
|
|
requires = ["hatchling"]
|
|
build-backend = "hatchling.build"
|