diff --git a/pkgs/development/python-modules/peacasso/default.nix b/pkgs/development/python-modules/peacasso/default.nix new file mode 100644 index 000000000000..2d4d5c0564a9 --- /dev/null +++ b/pkgs/development/python-modules/peacasso/default.nix @@ -0,0 +1,67 @@ +{ + lib, + buildPythonPackage, + fetchPypi, + setuptools, + setuptools-scm, + accelerate, + diffusers, + fastapi, + ftfy, + pydantic, + scipy, + torch, + transformers, + typer, + uvicorn, + nix-update-script, +}: + +buildPythonPackage rec { + pname = "peacasso"; + version = "0.0.19a0"; + pyproject = true; + + # No releases or tags are available in https://github.com/victordibia/peacasso + src = fetchPypi { + inherit pname version; + hash = "sha256-qBoG9FAJs0oZrQ0jShtPZfZPmyUZD30MGXDUfMl5bQk="; + }; + + build-system = [ + setuptools + setuptools-scm + ]; + + dependencies = [ + accelerate + diffusers + fastapi + ftfy + pydantic + scipy + torch + transformers + typer + uvicorn + ]; + + pythonRelaxDeps = [ "diffusers" ]; + + pythonImportsCheck = [ + "peacasso" + ]; + + # No tests + doCheck = false; + + passthru.updateScript = nix-update-script { }; + + meta = { + description = "UI tool to help you generate art (and experiment) with multimodal (text, image) AI models (stable diffusion)"; + homepage = "https://github.com/victordibia/peacasso"; + license = lib.licenses.mit; + maintainers = with lib.maintainers; [ moraxyc ]; + mainProgram = "peacasso"; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index ed176858c674..64b0b00c8a7b 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -10303,6 +10303,8 @@ self: super: with self; { pdoc3 = callPackage ../development/python-modules/pdoc3 { }; + peacasso = callPackage ../development/python-modules/peacasso { }; + peaqevcore = callPackage ../development/python-modules/peaqevcore { }; pegen = callPackage ../development/python-modules/pegen { };