From d12a51a9d5d14253c17fb76ae829a516b4f84805 Mon Sep 17 00:00:00 2001
From: Denis Manherz <denis@manherz.de>
Date: Sat, 2 Dec 2023 01:45:55 +0100
Subject: [PATCH] Update

---
 packages/xformers/default.nix      | 4 ++--
 projects/automatic1111/package.nix | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/packages/xformers/default.nix b/packages/xformers/default.nix
index f0a2eae..4040463 100644
--- a/packages/xformers/default.nix
+++ b/packages/xformers/default.nix
@@ -2,7 +2,7 @@
 # If you run pynixify again, the file will be either overwritten or
 # deleted, and you will lose the changes you made to it.
 
-{ buildPythonPackage, fetchPypi, lib, torch, numpy, pyre-extensions, pythonRelaxDepsHook, which }:
+{ buildPythonPackage, fetchPypi, lib, torchWithCuda, numpy, pyre-extensions, pythonRelaxDepsHook, which }:
 
 buildPythonPackage rec {
   pname = "xformers";
@@ -14,7 +14,7 @@ buildPythonPackage rec {
   };
   nativeBuildInputs = [ pythonRelaxDepsHook which ];
   pythonRelaxDeps = [ "pyre-extensions" ];
-  propagatedBuildInputs = [ torch numpy pyre-extensions /*triton*/ ];
+  propagatedBuildInputs = [ torchWithCuda numpy pyre-extensions /*triton*/ ];
 
   # TODO FIXME
   doCheck = false;
diff --git a/projects/automatic1111/package.nix b/projects/automatic1111/package.nix
index a487da1..8b8f224 100644
--- a/projects/automatic1111/package.nix
+++ b/projects/automatic1111/package.nix
@@ -36,7 +36,7 @@ python3Packages.buildPythonPackage {
     taming-transformers-rom1504
     timm
     tomesd
-    torch
+    torchWithCuda
     transformers
     xformers