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