pyrox.dev
1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5 regex,
6 requests,
7 setuptools,
8 tqdm,
9 numpy,
10 toposort,
11 tensorflow,
12}:
13
14buildPythonPackage rec {
15 pname = "gpt-2-simple";
16 version = "0.8.1";
17 pyproject = true;
18
19 src = fetchFromGitHub {
20 owner = "minimaxir";
21 repo = "gpt-2-simple";
22 rev = "v${version}";
23 hash = "sha256-WwD4sDcc28zXEOISJsq8e+rgaNrrgIy79Wa4J3E7Ovc=";
24 };
25
26 build-system = [ setuptools ];
27
28 propagatedBuildInputs = [
29 regex
30 requests
31 tqdm
32 numpy
33 toposort
34 tensorflow
35 ];
36
37 doCheck = false; # no tests in upstream
38
39 meta = with lib; {
40 description = "Easily retrain OpenAI's GPT-2 text-generating model on new texts";
41 homepage = "https://github.com/minimaxir/gpt-2-simple";
42 license = licenses.mit;
43 maintainers = [ ];
44 };
45}