This commit is contained in:
Denis Manherz 2024-08-11 19:17:02 +02:00
parent a227d57d28
commit 7fc90fd614

View file

@ -1,46 +1,19 @@
{ lib {
, buildPythonPackage buildPythonPackage,
, pkg-config pkg-config,
, fetchPypi stdenv,
, stdenv fetchzip,
, fetchzip setuptools,
, setuptools isPy27,
, writeText numpy,
, isPy27 opencv4,
, pytestCheckHook pillow,
, pytest-mpl symlinkJoin,
, numpy }: let
, scipy
, scikit-learn
, pandas
, transformers
, opencv4
, lightgbm
, catboost
, pyspark
, sentencepiece
, tqdm
, slicer
, numba
, matplotlib
, nose
, lime
, cloudpickle
, ipython
, packaging
, pillow
, requests
, regex
, importlib-metadata
, huggingface-hub
, symlinkJoin
}:
let
# https://github.com/invoke-ai/InvokeAI/blob/34f8117241dc961e78929bda30ce3b4f19e707cf/docs/installation/060_INSTALL_PATCHMATCH.md # https://github.com/invoke-ai/InvokeAI/blob/34f8117241dc961e78929bda30ce3b4f19e707cf/docs/installation/060_INSTALL_PATCHMATCH.md
opencv4Fixed = symlinkJoin { opencv4Fixed = symlinkJoin {
name = "opencv4Fixed"; name = "opencv4Fixed";
paths = [ opencv4 ]; paths = [opencv4];
postBuild = '' postBuild = ''
cp -r $out/lib/pkgconfig/opencv4.pc $out/lib/pkgconfig/opencv.pc cp -r $out/lib/pkgconfig/opencv4.pc $out/lib/pkgconfig/opencv.pc
''; '';
@ -62,31 +35,32 @@ let
cp libpatchmatch.so $out/lib/ cp libpatchmatch.so $out/lib/
''; '';
}; };
in buildPythonPackage { in
pname = "pypatchmatch"; buildPythonPackage {
version = "129863937a8ab37f6bbcec327c994c0f932abdbc"; pname = "pypatchmatch";
version = "129863937a8ab37f6bbcec327c994c0f932abdbc";
disabled = isPy27; disabled = isPy27;
src = fetchzip { src = fetchzip {
url = "https://github.com/invoke-ai/PyPatchMatch/archive/129863937a8ab37f6bbcec327c994c0f932abdbc.zip"; url = "https://github.com/invoke-ai/PyPatchMatch/archive/129863937a8ab37f6bbcec327c994c0f932abdbc.zip";
sha256 = "sha256-kHYih9fjhtYfyNYzW4kwzm62N+GaOQQOOlSkO4PH3lw="; sha256 = "sha256-kHYih9fjhtYfyNYzW4kwzm62N+GaOQQOOlSkO4PH3lw=";
}; };
propagatedBuildInputs = [ propagatedBuildInputs = [
setuptools setuptools
numpy numpy
pillow pillow
]; ];
doCheck = false; doCheck = false;
postInstall = '' postInstall = ''
cp ${libpatchmatch}/lib/libpatchmatch.so $out/lib/*/site-packages/patchmatch/ cp ${libpatchmatch}/lib/libpatchmatch.so $out/lib/*/site-packages/patchmatch/
''; '';
meta = { meta = {
description = "This library implements the PatchMatch based inpainting algorithm."; description = "This library implements the PatchMatch based inpainting algorithm.";
homepage = "https://github.com/invoke-ai/PyPatchMatch"; homepage = "https://github.com/invoke-ai/PyPatchMatch";
}; };
} }