pkgs/development/python-modules/gpt-2-simple/default.nix

   1 {
   2   lib,
   3   buildPythonPackage,
   4   fetchFromGitHub,
   5   regex,
   6   requests,
   7   setuptools,
   8   tqdm,
   9   numpy,
  10   toposort,
  11   tensorflow,
  12 }:
  13
  14 buildPythonPackage 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 }