dev/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

python311Packages.dalle-mini: fix build

Browse Source
This commit is contained in:
Gaetan Lepage 2023-08-11 20:00:38 +02:00
parent 43a394b99a
commit bf23d67430
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
pkgs/development/python-modules/dalle-mini/default.nix
View File

@ -1,6 +1,7 @@
{ lib { lib
, buildPythonPackage , buildPythonPackage
, fetchPypi , fetchPypi
, fetchpatch
, einops , einops
, emoji , emoji
, flax , flax
@ -16,16 +17,20 @@
buildPythonPackage rec { buildPythonPackage rec {
pname = "dalle-mini"; pname = "dalle-mini";
version = "0.1.5"; version = "0.1.5";
format = "setuptools";
src = fetchPypi { src = fetchPypi {
inherit pname version; inherit pname version;
hash = "sha256-k4XILjNNz0FPcAzwPEeqe5Lj24S2Y139uc9o/1IUS1c="; hash = "sha256-k4XILjNNz0FPcAzwPEeqe5Lj24S2Y139uc9o/1IUS1c=";
}; };
format = "setuptools"; # Fix incompatibility with the latest JAX versions
# See https://github.com/borisdayma/dalle-mini/pull/338
buildInputs = [ patches = [
jaxlib (fetchpatch {
url = "https://github.com/borisdayma/dalle-mini/pull/338/commits/22ffccf03f3e207731a481e3e42bdb564ceebb69.patch";
hash = "sha256-LIOyfeq/oVYukG+1rfy5PjjsJcjADCjn18x/hVmLkPY=";
})
]; ];
propagatedBuildInputs = [ propagatedBuildInputs = [
@ -34,6 +39,7 @@ buildPythonPackage rec {
flax flax
ftfy ftfy
jax jax
jaxlib
pillow pillow
transformers transformers
unidecode unidecode
@ -49,7 +55,5 @@ buildPythonPackage rec {
homepage = "https://github.com/borisdayma/dalle-mini"; homepage = "https://github.com/borisdayma/dalle-mini";
license = licenses.asl20; license = licenses.asl20;
maintainers = with maintainers; [ r-burns ]; maintainers = with maintainers; [ r-burns ];
# incompatible with recent versions of JAX
broken = true;
}; };
} }