tjh.dev
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
16buildPythonPackage rec {
17 pname = "dalle-mini";
18 version = "0.1.4";
19
20 src = fetchPypi {
21 inherit pname version;
22 hash = "sha256-UwCcoKbGxZT5XB+Mtv8kAHFdj0iLw8U1Ayo60y3Tm7U=";
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}