creek

The AI Image Editor of 2030
git clone https://github.com/maydayv7/creek

shell.nix (1274B)

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
{
  pkgs ? import <nixpkgs> {
    config = {
      allowUnfree = true;
      cudaSupport = true;
    };
  },
}:

pkgs.mkShell {
  nativeBuildInputs =
    let
      fal-client =
        with pkgs.python3Packages;
        buildPythonPackage rec {
          pname = "fal_client";
          version = "0.10.0";
          pyproject = true;
          src = fetchPypi {
            inherit pname version;
            hash = "sha256-UwNtAwgRerLaddWLTQnpUQk2Oob3sW/FmhhidgKYZh8=";
          };
          propagatedBuildInputs = [
            httpx
            httpx-sse
            msgpack
            setuptools
            setuptools-scm
            websockets
          ];
        };
    in
    [
      pkgs.python3
      (pkgs.python3.withPackages (
        p: with p; [
          accelerate
          bitsandbytes
          diffusers
          einops
          fal-client
          flask
          flask-cors
          hf-xet
          kornia
          numpy
          opencv4
          pillow
          pip
          pycryptodome
          python-dotenv
          requests
          scipy
          tblib
          timm
          toml
          torch
          torchvision
          transformers
        ]
      ))
    ];

  LD_LIBRARY_PATH = "${pkgs.stdenv.cc.cc.lib}/lib";
}