feat(exo): compile w/o CUDA

rafaelsgirao · Jan 29, 2025 · b49a2d7 · b49a2d7
1 parent 967c3af
commit b49a2d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/exo/default.nix b/packages/exo/default.nix
@@ -46,7 +46,7 @@ python3Packages.buildPythonApplication {
     transformers
     tinygrad
     # tensorflow #tensorflow not supported in python3.12
-    torchWithCuda
+    torchWithoutCuda
     # flax - tensorflow not supported in python3.12
   ];