mirror of
https://github.com/elder-plinius/OBLITERATUS.git
synced 2026-04-23 03:36:25 +02:00
GangGreenTemperTatum
67 lines
1.7 KiB
TOML
67 lines
1.7 KiB
TOML
[build-system]
|
|
requires = ["setuptools>=68.0", "wheel"]
|
|
build-backend = "setuptools.build_meta"
|
|
|
|
[project]
|
|
name = "obliteratus"
|
|
version = "0.1.2"
|
|
description = "Master Ablation Suite for HuggingFace transformers"
|
|
readme = "README.md"
|
|
requires-python = ">=3.10"
|
|
license = {text = "AGPL-3.0-or-later"}
|
|
keywords = ["abliteration", "mechanistic-interpretability", "refusal-removal", "llm", "transformers", "alignment"]
|
|
classifiers = [
|
|
"Development Status :: 4 - Beta",
|
|
"Intended Audience :: Science/Research",
|
|
"License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)",
|
|
"Programming Language :: Python :: 3.10",
|
|
"Programming Language :: Python :: 3.11",
|
|
"Programming Language :: Python :: 3.12",
|
|
"Topic :: Scientific/Engineering :: Artificial Intelligence",
|
|
]
|
|
|
|
dependencies = [
|
|
"torch>=2.0",
|
|
"transformers>=4.40",
|
|
"datasets>=2.14",
|
|
"accelerate>=0.24",
|
|
"safetensors>=0.4",
|
|
"pyyaml>=6.0",
|
|
"rich>=13.0",
|
|
"matplotlib>=3.7",
|
|
"seaborn>=0.12",
|
|
"pandas>=2.0",
|
|
"numpy>=1.24",
|
|
"scikit-learn>=1.3",
|
|
"tqdm>=4.64",
|
|
"bitsandbytes>=0.46.1",
|
|
]
|
|
|
|
[project.urls]
|
|
"Homepage" = "https://github.com/elder-plinius/OBLITERATUS"
|
|
"Repository" = "https://github.com/elder-plinius/OBLITERATUS"
|
|
"Bug Tracker" = "https://github.com/elder-plinius/OBLITERATUS/issues"
|
|
|
|
[project.optional-dependencies]
|
|
dev = ["pytest>=7.0", "pytest-cov", "ruff", "mypy"]
|
|
spaces = ["gradio>=5.0,<6.0"]
|
|
|
|
[project.scripts]
|
|
obliteratus = "obliteratus.cli:main"
|
|
|
|
[tool.setuptools.packages.find]
|
|
include = ["obliteratus*"]
|
|
|
|
[tool.setuptools.package-data]
|
|
obliteratus = ["py.typed"]
|
|
|
|
[tool.ruff]
|
|
line-length = 100
|
|
target-version = "py310"
|
|
|
|
[tool.pytest.ini_options]
|
|
testpaths = ["tests"]
|
|
|
|
[tool.uv]
|
|
exclude-newer = "3 days"
|