Commit bf23d674 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python311Packages.dalle-mini: fix build

parent 43a394b9

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

+10 −6
Original line number Diff line number Diff line 
{ lib

, buildPythonPackage

, fetchPypi

, fetchpatch

, einops

, emoji

, flax
@@ -16,16 +17,20 @@ 
buildPythonPackage rec {

  pname = "dalle-mini";

  version = "0.1.5";

  format = "setuptools";



  src = fetchPypi {

    inherit pname version;

    hash = "sha256-k4XILjNNz0FPcAzwPEeqe5Lj24S2Y139uc9o/1IUS1c=";

  };



  format = "setuptools";



  buildInputs = [

    jaxlib

  # Fix incompatibility with the latest JAX versions

  # See https://github.com/borisdayma/dalle-mini/pull/338

  patches = [

    (fetchpatch {

      url = "https://github.com/borisdayma/dalle-mini/pull/338/commits/22ffccf03f3e207731a481e3e42bdb564ceebb69.patch";

      hash = "sha256-LIOyfeq/oVYukG+1rfy5PjjsJcjADCjn18x/hVmLkPY=";

    })

  ];



  propagatedBuildInputs = [
@@ -34,6 +39,7 @@ buildPythonPackage rec { 
    flax

    ftfy

    jax

    jaxlib

    pillow

    transformers

    unidecode
@@ -49,7 +55,5 @@ buildPythonPackage rec { 
    homepage = "https://github.com/borisdayma/dalle-mini";

    license = licenses.asl20;

    maintainers = with maintainers; [ r-burns ];

    # incompatible with recent versions of JAX

    broken = true;

  };

}