diff --git a/flake.nix b/flake.nix index 673f7021771c0..c72d8e4e7197d 100644 --- a/flake.nix +++ b/flake.nix @@ -80,6 +80,8 @@ buildFlags = [ "LLAMA_CUBLAS=1" ]; installPhase = '' + runHook preInstall + mkdir -p $out/bin # TODO(Green-Sky): add install target to Makefile, or wait for cmake support @@ -91,6 +93,8 @@ mv quantize $out/bin/ mv llama-bench $out/bin/ mv train-text-from-scratch $out/bin/ + + runHook postInstall ''; }; packages.rocm = pkgs.stdenv.mkDerivation {