wtf is happening

InvokeAI

@@ -1 +1 @@
-Subproject commit 93cdb476d919b779e32a638aae15e56cc46b7adb
+Subproject commit 5beeb1a897c744a9faacf719e58dc5ac015a3d00

flake.nix

@@ -46,6 +46,14 @@
         pillow
       ]
       ++ nixlib.optional (!webui) [
+        npyscreen
+        huggingface-hub
+        dnspython
+        datasets
+        click
+        pypatchmatch
+        #torchsde
+        trampoline
         send2trash
         flask
         flask-socketio
@@ -149,6 +157,9 @@
           jsonmerge = callPackage ./packages/jsonmerge { };
           clean-fid = callPackage ./packages/clean-fid { };
           getpass-asterisk = callPackage ./packages/getpass-asterisk { };
+          pypatchmatch = callPackage ./packages/pypatchmatch { };
+          trampoline = callPackage ./packages/trampoline { };
+          torchsde = callPackage ./packages/torchsde { };
         };
       overlay_amd = nixpkgs: pythonPackages:
         rec {

packages/k-diffusion/default.nix

@@ -8,11 +8,11 @@
 
 buildPythonPackage rec {
   pname = "k-diffusion";
-  version = "0.0.9";
+  version = "0.0.14";
 
   src = fetchPypi {
     inherit pname version;
-    sha256 = "02j7hkhdh57bkvc75xygj50a64dzdi44d1gsw4wjmvp9f7pllpfa";
+    sha256 = "sha256-b7mxIA64tClz96s78GCI+kBJf/umi2ClCNjDYlqjTrs=";
   };
 
   propagatedBuildInputs = [

packages/pypatchmatch/default.nix

@@ -0,0 +1,25 @@
+{ buildPythonPackage, lib, numpy,  fetchFromGitHub, pillow, tqdm, setuptools }:
+
+buildPythonPackage rec {
+  pname = "pypatchmatch";
+  version = "0.1.1";
+
+  src = fetchFromGitHub {
+    owner = "invoke-ai";
+    repo = "PyPatchMatch";
+    rev = version;
+    sha256 = "sha256-PemWmujCMVzKF3/BL0jrL+z5KCKIertrYLAwf/+ZySs=";
+  };
+
+  format = "pyproject";
+  buildInputs = [ setuptools ];
+  propagatedBuildInputs = [ numpy pillow tqdm ];
+
+  # TODO FIXME
+  doCheck = false;
+
+  meta = {
+    description = "This library implements the PatchMatch based inpainting algorithm";
+    homepage = "https://github.com/invoke-ai/PyPatchMatch";
+  };
+}

packages/torchsde/default.nix

@@ -0,0 +1,26 @@
+# WARNING: This file was automatically generated. You should avoid editing it.
+# If you run pynixify again, the file will be either overwritten or
+# deleted, and you will lose the changes you made to it.
+
+{ boltons, buildPythonPackage, fetchPypi, lib, numpy, scipy, torch }:
+
+buildPythonPackage rec {
+  pname = "torchsde";
+  version = "0.2.5";
+
+  src = fetchPypi {
+    inherit pname version;
+    sha256 = "0swhs2rp1hqi7qm55xh2rbcrz3qp8b2a1kvib95pmlqhavhyjar2";
+  };
+
+  propagatedBuildInputs = [ boltons numpy scipy torch ];
+
+  # TODO FIXME
+  doCheck = false;
+
+  meta = with lib; {
+    description =
+      "SDE solvers and stochastic adjoint sensitivity analysis in PyTorch.";
+    homepage = "https://github.com/google-research/torchsde";
+  };
+}

packages/trampoline/default.nix

@@ -0,0 +1,29 @@
+# WARNING: This file was automatically generated. You should avoid editing it.
+# If you run pynixify again, the file will be either overwritten or
+# deleted, and you will lose the changes you made to it.
+
+{ boltons, buildPythonPackage, fetchPypi, lib, numpy, scipy, torch, fetchFromGitLab }:
+
+buildPythonPackage rec {
+  pname = "trampoline";
+  version = "0.1.2";
+
+  src = fetchFromGitLab {
+    inherit pname version;
+    owner = "ferreum";
+    repo = pname;
+    rev = "6ff003ed89abc4b64587227d10a6a8ba48309a83";
+    sha256 = "";
+  };
+
+  propagatedBuildInputs = [ ];
+
+  # TODO FIXME
+  doCheck = false;
+
+  meta = with lib; {
+    description =
+      "This trampoline allows recursive functions to recurse virtually (or literally) infinitely. Most existing recursive functions can be converted with simple modifications.";
+    homepage = "https://pypi.org/project/trampoline/#description";
+  };
+}