diff --git a/pkgs/development/python-modules/gliner/default.nix b/pkgs/development/python-modules/gliner/default.nix new file mode 100644 index 000000000000..77583b257065 --- /dev/null +++ b/pkgs/development/python-modules/gliner/default.nix @@ -0,0 +1,60 @@ +{ + lib, + buildPythonPackage, + fetchFromGitHub, + + # build-system + setuptools, + + # dependencies + huggingface-hub, + onnxruntime, + sentencepiece, + torch, + tqdm, + transformers, +}: + +buildPythonPackage rec { + pname = "gliner"; + version = "0.2.20"; + pyproject = true; + + src = fetchFromGitHub { + owner = "urchade"; + repo = "GLiNER"; + tag = "v${version}"; + hash = "sha256-aWBDnaiq9Z30YT4sszEVk1WAyU4aH8SFD6ESOBkT2ds="; + }; + + build-system = [ + setuptools + ]; + + dependencies = [ + huggingface-hub + onnxruntime + sentencepiece + torch + tqdm + transformers + ]; + + pythonImportsCheck = [ "gliner" ]; + + # All tests require internet + doCheck = false; + + meta = { + description = "Generalist and Lightweight Model for Named Entity Recognition"; + homepage = "https://github.com/urchade/GLiNER"; + changelog = "https://github.com/urchade/GLiNER/releases/tag/v${version}"; + license = lib.licenses.asl20; + maintainers = with lib.maintainers; [ GaetanLepage ]; + badPlatforms = [ + # terminate called after throwing an instance of 'onnxruntime::OnnxRuntimeException' + # Attempt to use DefaultLogger but none has been registered. + "aarch64-linux" + ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 85461c1feb53..b75310f7202b 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -5688,6 +5688,8 @@ self: super: with self; { glfw = callPackage ../development/python-modules/glfw { }; + gliner = callPackage ../development/python-modules/gliner { }; + glob2 = callPackage ../development/python-modules/glob2 { }; globus-sdk = callPackage ../development/python-modules/globus-sdk { };