Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.4 kB · 85 lines
Nix
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586{ lib, fetchFromGitHub, buildPythonPackage, substituteAll
# build-system, setuptools
# runtime, ffmpeg-headless
# propagates, more-itertools, numba, numpy, openai-triton, scipy, tiktoken, torch, tqdm, transformers
# tests, pytestCheckHook}:
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 openai-triton scipy tiktoken torch tqdm transformers ];
preCheck = '' export HOME=$TMPDIR '';
nativeCheckInputs = [ pytestCheckHook ];
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"; homepage = "https://github.com/openai/whisper"; license = licenses.mit; maintainers = with maintainers; [ hexa MayNiklas ]; };}