61 lines
1,007 B
Nix
61 lines
1,007 B
Nix
{
|
|
lib,
|
|
buildPythonPackage,
|
|
fetchFromGitHub,
|
|
setuptools,
|
|
setuptools-scm,
|
|
interegular,
|
|
cloudpickle,
|
|
diskcache,
|
|
joblib,
|
|
jsonschema,
|
|
pydantic,
|
|
lark,
|
|
nest-asyncio,
|
|
numba,
|
|
scipy,
|
|
torch,
|
|
transformers,
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "outlines";
|
|
version = "0.0.38";
|
|
pyproject = true;
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "outlines-dev";
|
|
repo = pname;
|
|
rev = "refs/tags/${version}";
|
|
hash = "sha256-kOd2J6PV2idWaltkGUVNXfB+1dNU1f8vbnHQ5i1WxG8=";
|
|
};
|
|
|
|
nativeBuildInputs = [
|
|
setuptools
|
|
setuptools-scm
|
|
];
|
|
|
|
propagatedBuildInputs = [
|
|
interegular
|
|
cloudpickle
|
|
diskcache
|
|
joblib
|
|
jsonschema
|
|
pydantic
|
|
lark
|
|
nest-asyncio
|
|
numba
|
|
scipy
|
|
torch
|
|
transformers
|
|
];
|
|
|
|
pythonImportsCheck = [ "outlines" ];
|
|
|
|
meta = with lib; {
|
|
description = "Structured text generation";
|
|
homepage = "https://github.com/outlines-dev/outlines";
|
|
license = licenses.asl20;
|
|
maintainers = with maintainers; [ lach ];
|
|
};
|
|
}
|