pkgs/development/python-modules/dalle-mini/default.nix

   1 { lib
   2 , buildPythonPackage
   3 , fetchPypi
   4 , einops
   5 , emoji
   6 , flax
   7 , ftfy
   8 , jax
   9 , jaxlib
  10 , pillow
  11 , transformers
  12 , unidecode
  13 , wandb
  14 }:
  15
  16 buildPythonPackage rec {
  17   pname = "dalle-mini";
  18   version = "0.1.1";
  19
  20   src = fetchPypi {
  21     inherit pname version;
  22     sha256 = "sha256-/wGIuYSWEUgJmeRN5K9/xuoCs+hpFX4/Tu1un1C4ljk=";
  23   };
  24
  25   format = "setuptools";
  26
  27   buildInputs = [
  28     jaxlib
  29   ];
  30
  31   propagatedBuildInputs = [
  32     einops
  33     emoji
  34     flax
  35     ftfy
  36     jax
  37     pillow
  38     transformers
  39     unidecode
  40     wandb
  41   ];
  42
  43   doCheck = false; # no upstream tests
  44
  45   pythonImportsCheck = [ "dalle_mini" ];
  46
  47   meta = with lib; {
  48     description = "Generate images from a text prompt";
  49     homepage = "https://github.com/borisdayma/dalle-mini";
  50     license = licenses.asl20;
  51     maintainers = with maintainers; [ r-burns ];
  52   };
  53 }