add python311Packages.io-paint-cuda

jpetrucciani · Mar 14, 2024 · 2023baf · 2023baf
1 parent d16e5ed
commit 2023baf
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/mods/python/ai/apps.nix b/mods/python/ai/apps.nix
@@ -83,9 +83,8 @@ let
         maintainers = with maintainers; [ jpetrucciani ];
       };
     };
-in
-{
-  io-paint = buildPythonPackage rec {
+
+  _io-paint = buildPythonPackage rec {
     pname = "io-paint";
     version = "1.2.2";
     pyproject = true;
@@ -141,6 +140,10 @@ in
     };
   };
 
+in
+{
+  io-paint = _io-paint;
+  io-paint-cuda = _io-paint.override { torch = final.torchWithCuda; };
   chainforge = buildPythonPackage rec {
     pname = "chainforge";
     version = "0.2.5.3";