shuffled dependencies a bit

2024-10-26 03:52:40 +03:00 · 2022-11-15 23:28:13 +10:00 · 2022-11-15 23:28:13 +10:00 · 68898ab07b
commit 68898ab07b
parent b50c596d45
1 changed files with 12 additions and 9 deletions
--- a/flake.nix
+++ b/flake.nix
@ -23,7 +23,6 @@
        torchvision
        numpy

-        # TODO: split SD and InvokeAI deps
        albumentations
        opencv4
        pudb
@ -32,26 +31,30 @@
        pytorch-lightning
        protobuf3_20
        omegaconf
-        realesrgan
        test-tube
        streamlit
-        send2trash
-        pillow
        einops
        taming-transformers-rom1504
        torch-fidelity
        torchmetrics
        transformers
+        kornia
+        k-diffusion
+
+        # following packages not needed for vanilla SD but used by both UIs
+        realesrgan
+        pillow
+      ] 
+      ++ nixlib.optional (!webui) [
+        send2trash
        flask
        flask-socketio
        flask-cors
        dependency-injector
-        eventlet
-        kornia
-        clip
-        k-diffusion
        gfpgan
-      ] ++ nixlib.optional webui [
+        eventlet
+      ]
+      ++ nixlib.optional webui [
        addict
        future
        lmdb