53 lines
934 B
Nix
53 lines
934 B
Nix
{ lib
|
|
, buildPythonPackage
|
|
, fetchPypi
|
|
, setuptools
|
|
, wheel
|
|
, filelock
|
|
, huggingface-hub
|
|
, opencv-python
|
|
, torchvision
|
|
, einops
|
|
, scikit-image
|
|
, timm
|
|
, pythonRelaxDepsHook
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "controlnet-aux";
|
|
version = "0.0.7";
|
|
format = "pyproject";
|
|
|
|
src = fetchPypi {
|
|
pname = "controlnet_aux";
|
|
inherit version;
|
|
hash = "sha256-23KZMjum04ni/mt9gTGgWica86SsKldHdUSMTQd4vow=";
|
|
};
|
|
|
|
propagatedBuildInputs = [
|
|
filelock
|
|
huggingface-hub
|
|
opencv-python
|
|
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; [ ];
|
|
};
|
|
}
|