63 lines
994 B
Nix
63 lines
994 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 ];
|
||
|
};
|
||
|
}
|