tjh.dev
1{
2 lib,
3 buildPythonPackage,
4 fetchPypi,
5 setuptools,
6 setuptools-scm,
7 accelerate,
8 diffusers,
9 fastapi,
10 ftfy,
11 pydantic,
12 scipy,
13 torch,
14 transformers,
15 typer,
16 uvicorn,
17 nix-update-script,
18}:
19
20buildPythonPackage rec {
21 pname = "peacasso";
22 version = "0.0.19a0";
23 pyproject = true;
24
25 # No releases or tags are available in https://github.com/victordibia/peacasso
26 src = fetchPypi {
27 inherit pname version;
28 hash = "sha256-qBoG9FAJs0oZrQ0jShtPZfZPmyUZD30MGXDUfMl5bQk=";
29 };
30
31 build-system = [
32 setuptools
33 setuptools-scm
34 ];
35
36 dependencies = [
37 accelerate
38 diffusers
39 fastapi
40 ftfy
41 pydantic
42 scipy
43 torch
44 transformers
45 typer
46 uvicorn
47 ];
48
49 pythonRelaxDeps = [ "diffusers" ];
50
51 pythonImportsCheck = [
52 "peacasso"
53 ];
54
55 # No tests
56 doCheck = false;
57
58 passthru.updateScript = nix-update-script { };
59
60 meta = {
61 description = "UI tool to help you generate art (and experiment) with multimodal (text, image) AI models (stable diffusion)";
62 homepage = "https://github.com/victordibia/peacasso";
63 license = lib.licenses.mit;
64 maintainers = with lib.maintainers; [ moraxyc ];
65 mainProgram = "peacasso";
66 };
67}