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

   1 {
   2   lib,
   3   buildPythonPackage,
   4   fetchPypi,
   5   fetchpatch,
   6   einops,
   7   emoji,
   8   flax,
   9   ftfy,
  10   jax,
  11   jaxlib,
  12   pillow,
  13   transformers,
  14   unidecode,
  15   wandb,
  16 }:
  17
  18 buildPythonPackage rec {
  19   pname = "dalle-mini";
  20   version = "0.1.5";
  21   format = "setuptools";
  22
  23   src = fetchPypi {
  24     inherit pname version;
  25     hash = "sha256-k4XILjNNz0FPcAzwPEeqe5Lj24S2Y139uc9o/1IUS1c=";
  26   };
  27
  28   # Fix incompatibility with the latest JAX versions
  29   # See https://github.com/borisdayma/dalle-mini/pull/338
  30   patches = [
  31     (fetchpatch {
  32       url = "https://github.com/borisdayma/dalle-mini/pull/338/commits/22ffccf03f3e207731a481e3e42bdb564ceebb69.patch";
  33       hash = "sha256-LIOyfeq/oVYukG+1rfy5PjjsJcjADCjn18x/hVmLkPY=";
  34     })
  35   ];
  36
  37   propagatedBuildInputs = [
  38     einops
  39     emoji
  40     flax
  41     ftfy
  42     jax
  43     jaxlib
  44     pillow
  45     transformers
  46     unidecode
  47     wandb
  48   ];
  49
  50   doCheck = false; # no upstream tests
  51
  52   pythonImportsCheck = [ "dalle_mini" ];
  53
  54   meta = with lib; {
  55     description = "Generate images from a text prompt";
  56     homepage = "https://github.com/borisdayma/dalle-mini";
  57     license = licenses.asl20;
  58     maintainers = with maintainers; [ r-burns ];
  59   };
  60 }