diff --git a/flake.nix b/flake.nix
index 723848c..368b0a7 100644
--- a/flake.nix
+++ b/flake.nix
@@ -17,6 +17,8 @@
           safetensors
           huggingface-hub
           numpy
+          tokenizers
+          datasets
         ]);
 
         colibri = pkgs.stdenv.mkDerivation {