blob: ee78451aae8d42a1125df2bdfaa55008210b6662 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
{ lib, stdenv, fetchurl, autoPatchelfHook, python3 }:
stdenv.mkDerivation rec {
pname = "gpt2tc";
version = "2021-04-24";
src = fetchurl {
url = "https://web.archive.org/web/20220603034455/https://bellard.org/libnc/gpt2tc-2021-04-24.tar.gz";
hash = "sha256-6oTxnbBwjHAXVrWMjOQVwdODbqLRoinx00pi29ff5w0=";
};
patches = [
# Add a shebang to the python script so that nix detects it as such and
# wraps it properly. Otherwise, it runs in shell and freezes the system.
./0001-add-python-shebang.patch
];
nativeBuildInputs = [ autoPatchelfHook ];
buildInputs = [
(python3.withPackages (p: with p; [ numpy tensorflow ]))
];
installPhase = ''
runHook preInstall
install -D -m755 -t $out/lib libnc${stdenv.hostPlatform.extensions.sharedLibrary}
addAutoPatchelfSearchPath $out/lib
install -D -m755 -t $out/bin gpt2tc
install -T -m755 download_model.sh $out/bin/gpt2-download-model
install -T -m755 gpt2convert.py $out/bin/gpt2-convert
install -D -m644 -t $out/share/gpt2tc readme.txt gpt2vocab.txt Changelog
runHook postInstall
'';
meta = with lib; {
description = "Text completion and compression using GPT-2";
homepage = "https://bellard.org/libnc/gpt2tc.html";
sourceProvenance = with sourceTypes; [ binaryNativeCode ];
license = licenses.unfree;
platforms = [ "x86_64-linux" ];
maintainers = with maintainers; [ anna328p ];
};
}
|
