pyrox.dev
1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5
6 # build-system
7 setuptools,
8 setuptools-scm,
9
10 # dependencies
11 accelerate,
12 datasets,
13 rich,
14 transformers,
15}:
16
17buildPythonPackage rec {
18 pname = "trl";
19 version = "0.23.0";
20 pyproject = true;
21
22 src = fetchFromGitHub {
23 owner = "huggingface";
24 repo = "trl";
25 tag = "v${version}";
26 hash = "sha256-LHBFcf1araJ9Eosrr2z87HrRt5en1jIZ9C9pjqi9Nik=";
27 };
28
29 build-system = [
30 setuptools
31 setuptools-scm
32 ];
33
34 dependencies = [
35 accelerate
36 datasets
37 rich
38 transformers
39 ];
40
41 # Many tests require internet access.
42 doCheck = false;
43
44 pythonImportsCheck = [ "trl" ];
45
46 meta = {
47 description = "Train transformer language models with reinforcement learning";
48 homepage = "https://github.com/huggingface/trl";
49 changelog = "https://github.com/huggingface/trl/releases/tag/${src.tag}";
50 license = lib.licenses.asl20;
51 maintainers = with lib.maintainers; [ hoh ];
52 };
53}