summary refs log tree commit diff stats
path: root/classification/shell.nix
diff options
context:
space:
mode:
Diffstat (limited to 'classification/shell.nix')
-rw-r--r--classification/shell.nix1
1 files changed, 1 insertions, 0 deletions
diff --git a/classification/shell.nix b/classification/shell.nix
index 5a20b8e46..5a992f0f8 100644
--- a/classification/shell.nix
+++ b/classification/shell.nix
@@ -4,6 +4,7 @@ mkShell {
   buildInputs = [
     python312Packages.transformers
     python312Packages.torch
+    python312Packages.ollama
   ];
   shellHook = ''
     # fixes libstdc++ issues and libgl.so issues