pyrox.dev
1{
2 lib,
3 buildPythonPackage,
4 catalogue,
5 curated-tokenizers,
6 fetchFromGitHub,
7 huggingface-hub,
8 setuptools,
9 tokenizers,
10 torch,
11}:
12
13buildPythonPackage rec {
14 pname = "curated-transformers";
15 version = "2.0.1";
16 pyproject = true;
17
18 src = fetchFromGitHub {
19 owner = "explosion";
20 repo = "curated-transformers";
21 tag = "v${version}";
22 hash = "sha256-2sedBVpwCppviWix+d3tJFTrLBe+2IBlWnCKgV6MucA=";
23 };
24
25 build-system = [ setuptools ];
26
27 dependencies = [
28 catalogue
29 curated-tokenizers
30 huggingface-hub
31 tokenizers
32 torch
33 ];
34
35 # Unit tests are hard to use, since most tests rely on downloading
36 # models from Hugging Face Hub.
37 pythonImportsCheck = [ "curated_transformers" ];
38
39 meta = {
40 description = "PyTorch library of curated Transformer models and their composable components";
41 homepage = "https://github.com/explosion/curated-transformers";
42 changelog = "https://github.com/explosion/curated-transformers/releases/tag/v${version}";
43 license = lib.licenses.mit;
44 maintainers = with lib.maintainers; [ danieldk ];
45 };
46}