55 lines
976 B
Nix
55 lines
976 B
Nix
{
|
|
lib,
|
|
buildPythonPackage,
|
|
fetchPypi,
|
|
setuptools,
|
|
wheel,
|
|
filelock,
|
|
huggingface-hub,
|
|
opencv4,
|
|
torchvision,
|
|
einops,
|
|
scikit-image,
|
|
timm,
|
|
importlib-metadata,
|
|
pythonRelaxDepsHook,
|
|
}:
|
|
buildPythonPackage rec {
|
|
pname = "controlnet-aux";
|
|
version = "0.0.7";
|
|
format = "pyproject";
|
|
|
|
src = fetchPypi {
|
|
pname = "controlnet_aux";
|
|
inherit version;
|
|
hash = "sha256-23KZMjum04ni/mt9gTGgWica86SsKldHdUSMTQd4vow=";
|
|
};
|
|
|
|
propagatedBuildInputs = [
|
|
importlib-metadata
|
|
filelock
|
|
huggingface-hub
|
|
opencv4
|
|
torchvision
|
|
einops
|
|
scikit-image
|
|
timm
|
|
];
|
|
|
|
nativeBuildInputs = [
|
|
setuptools
|
|
wheel
|
|
pythonRelaxDepsHook
|
|
];
|
|
|
|
pythonImportsCheck = ["controlnet_aux"];
|
|
pythonRemoveDeps = ["opencv-python"];
|
|
|
|
meta = with lib; {
|
|
description = "Auxillary models for controlnet";
|
|
homepage = "https://pypi.org/project/controlnet-aux/";
|
|
license = licenses.asl20;
|
|
maintainers = with maintainers; [];
|
|
};
|
|
}
|