Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.5 kB · 86 lines
Nix
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687{ lib, stdenv, fetchFromGitHub, buildPythonPackage, substituteAll,
# build-system setuptools,
# runtime ffmpeg-headless,
# propagates more-itertools, numba, numpy, triton, tiktoken, torch, tqdm,
# tests pytestCheckHook, scipy,}:
buildPythonPackage rec { pname = "whisper"; version = "20231117"; pyproject = true;
src = fetchFromGitHub { owner = "openai"; repo = pname; rev = "refs/tags/v${version}"; hash = "sha256-MJ1XjB/GuYUiECCuuHS0NWHvvs+ko0oTvLuDI7zLNiY="; };
patches = [ (substituteAll { src = ./ffmpeg-path.patch; ffmpeg = ffmpeg-headless; }) ];
nativeBuildInputs = [ setuptools ];
propagatedBuildInputs = [ more-itertools numba numpy tiktoken torch tqdm ] ++ lib.optionals (lib.meta.availableOn stdenv.hostPlatform triton) [ triton ];
preCheck = '' export HOME=$TMPDIR '';
nativeCheckInputs = [ pytestCheckHook scipy ];
disabledTests = [ # requires network access to download models "test_transcribe" # requires NVIDIA drivers "test_dtw_cuda_equivalence" "test_median_filter_equivalence" ];
meta = with lib; { changelog = "https://github.com/openai/whisper/blob/v${version}/CHANGELOG.md"; description = "General-purpose speech recognition model"; mainProgram = "whisper"; homepage = "https://github.com/openai/whisper"; license = licenses.mit; maintainers = with maintainers; [ hexa MayNiklas ]; };}