nixpkgs/pkgs/development/libraries/sentencepiece/default.nix
2023-05-04 03:41:02 +00:00

42 lines
1.0 KiB
Nix

{ lib
, fetchFromGitHub
, stdenv
, cmake
, gperftools
, withGPerfTools ? true
}:
stdenv.mkDerivation rec {
pname = "sentencepiece";
version = "0.1.99";
src = fetchFromGitHub {
owner = "google";
repo = pname;
rev = "refs/tags/v${version}";
sha256 = "sha256-RxzysZsfTdhAtJCO3JOa/bSBNnHBRWkqBdwBa8sB74I=";
};
nativeBuildInputs = [ cmake ];
buildInputs = lib.optionals withGPerfTools [ gperftools ];
outputs = [ "bin" "dev" "out" ];
# https://github.com/google/sentencepiece/issues/754
postPatch = ''
substituteInPlace CMakeLists.txt \
--replace '\$'{exec_prefix}/'$'{CMAKE_INSTALL_LIBDIR} '$'{CMAKE_INSTALL_FULL_LIBDIR} \
--replace '\$'{prefix}/'$'{CMAKE_INSTALL_INCLUDEDIR} '$'{CMAKE_INSTALL_FULL_INCLUDEDIR}
'';
meta = with lib; {
homepage = "https://github.com/google/sentencepiece";
description = "Unsupervised text tokenizer for Neural Network-based text generation";
license = licenses.asl20;
platforms = platforms.unix;
maintainers = with maintainers; [ pashashocky ];
};
}