From c3b955145998d39df39370671585a271ca6f80f0 Mon Sep 17 00:00:00 2001
From: Ben Sima <ben@bsima.me>
Date: Wed, 28 Dec 2022 19:53:55 -0500
Subject: Get ava GPT chatbot prototype working

Mostly thid required packaging up some deps, but also had to recompile
stuff with cuda support.
---
 Biz/Bild.nix | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

(limited to 'Biz/Bild.nix')

diff --git a/Biz/Bild.nix b/Biz/Bild.nix
index 47c0ecb..8fe20db 100644
--- a/Biz/Bild.nix
+++ b/Biz/Bild.nix
@@ -146,9 +146,15 @@ rec {
       haskell.packages.${constants.ghcCompiler}.fast-tags
       hlint
       lolcat
-      nixops
+      #nixops # fails to build
       ormolu
-      (python3.withPackages(p: with p; [transformers pytorch]))
+      (private.nixpkgs.python3.withPackages(p:
+      with p; [
+        transformers
+        pytorch
+        private.nixpkgs.python3Packages.bitsandbytes
+        private.nixpkgs.python3Packages.accelerate
+      ]))
       shellcheck
       wemux
     ];
-- 
cgit v1.2.3