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