Something went wrong. Try again.
lol
Something went wrong. Try again.
1.6 kB · 90 lines
Nix
at master
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091{ lib, accelerate, bitsandbytes, buildPythonPackage, datasets, fetchFromGitHub, # geom-median, hf-transfer, huggingface-hub, imageio, kernels, matplotlib, numpy, optuna, # pacmap, peft, psutil, pydantic-settings, questionary, rich, scikit-learn, transformers, uv-build,}:
buildPythonPackage (finalAttrs: { pname = "heretic-llm"; version = "1.2.0"; pyproject = true;
src = fetchFromGitHub { owner = "p-e-w"; repo = "heretic"; tag = "v${finalAttrs.version}"; hash = "sha256-KmqbOAOII1SP7wpdvGxtzQJt5NmlnF/o99NuZ21vO0s="; };
postPatch = '' substituteInPlace pyproject.toml \ --replace-fail "uv_build>=0.8.11,<0.9.0" "uv_build" '';
pythonRelaxDeps = [ "huggingface-hub" "transformers" ];
build-system = [ uv-build ];
dependencies = [ accelerate bitsandbytes datasets hf-transfer huggingface-hub kernels optuna peft psutil pydantic-settings questionary rich transformers ];
optional-dependencies = { research = [ # geom-median imageio matplotlib numpy # pacmap scikit-learn ]; };
pythonImportsCheck = [ "heretic" ];
meta = { description = "Tool to remove censorship removal for language models"; homepage = "https://github.com/p-e-w/heretic"; changelog = "https://github.com/p-e-w/heretic/releases/tag/v${finalAttrs.src.tag}"; license = with lib.licenses; [ agpl3Only agpl3Plus ]; maintainers = with lib.maintainers; [ fab ]; };})