Merge pull request #15 from atagen/master

fixed nvidia typo
gbtb · Dec 3, 2022 · 3f17c05 · 3f17c05
2 parents c9db788 + 1a6ff3b
commit 3f17c05
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/flake.nix b/flake.nix
@@ -177,9 +177,8 @@
  };
  overlay_nvidia = nixpkgs: pythonPackages:
  {
- torch = pythonPackages.torch.override {
- cudaSupport = true;
- };
+ torch = pythonPackages.torch-bin;
+ torchvision = pythonPackages.torchvision-bin;
  };
  in
  {
@@ -200,7 +199,7 @@
  python-self: python-super:
  (overlay_default prev python-super) //
  optional amd (overlay_amd prev python-super) //
- optional nvidia (overlay_amd prev python-super) //
+ optional nvidia (overlay_nvidia prev python-super) //
  optional webui (overlay_webui prev python-super) //
  (overlay_pynixify python-self);
  };