creek

 1{
 2  pkgs ? import <nixpkgs> {
 3    config = {
 4      allowUnfree = true;
 5      cudaSupport = true;
 6    };
 7  },
 8}:
 9
10pkgs.mkShell {
11  nativeBuildInputs =
12    let
13      fal-client =
14        with pkgs.python3Packages;
15        buildPythonPackage rec {
16          pname = "fal_client";
17          version = "0.10.0";
18          pyproject = true;
19          src = fetchPypi {
20            inherit pname version;
21            hash = "sha256-UwNtAwgRerLaddWLTQnpUQk2Oob3sW/FmhhidgKYZh8=";
22          };
23          propagatedBuildInputs = [
24            httpx
25            httpx-sse
26            msgpack
27            setuptools
28            setuptools-scm
29            websockets
30          ];
31        };
32    in
33    [
34      pkgs.python3
35      (pkgs.python3.withPackages (
36        p: with p; [
37          accelerate
38          bitsandbytes
39          diffusers
40          einops
41          fal-client
42          flask
43          flask-cors
44          hf-xet
45          kornia
46          numpy
47          opencv4
48          pillow
49          pip
50          pycryptodome
51          python-dotenv
52          requests
53          scipy
54          tblib
55          timm
56          toml
57          torch
58          torchvision
59          transformers
60        ]
61      ))
62    ];
63
64  LD_LIBRARY_PATH = "${pkgs.stdenv.cc.cc.lib}/lib";
65}