oops

nicball · Jan 22, 2024 · 8c55e54 · 8c55e54
1 parent ef4fa38
commit 8c55e54
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pkgs/all-packages.nix b/pkgs/all-packages.nix
@@ -90,6 +90,8 @@ self: super:
     modifyDerivationOutput
     wrapDerivationOutput;
 
-  torchvisionWithRocm = let p = self.python3Packages; in p.torchvision.override { torch = p.torchWithRocm; };
+  torchvisionWithRocm = let p = self.python3Packages; in p.torchvision.override { torch = p.torchWithRocm; } // {
+    meta.platforms = self.lib.platforms.x86;
+  };
 
 }