colin/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #318240 from mweinelt/whisper-ctranslate2-diarization

Browse Source 
whisper-ctranslate2: support speaker diarization
This commit is contained in:
Martin Weinelt 2024-06-08 17:11:23 +02:00 committed by GitHub
commit 225f52c561
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
pkgs/tools/audio/whisper-ctranslate2/default.nix
View File

@ -9,7 +9,7 @@ let
in
python3.pkgs.buildPythonApplication {
inherit pname version;
format = "setuptools";
pyproject = true;
disabled = python3.pythonOlder "3.6";
@ -20,25 +20,30 @@ python3.pkgs.buildPythonApplication {
hash = "sha256-iVS1wyPCXlbK1rMFidNbbUohu527NSaCpu1Dve01TvM=";
};
propagatedBuildInputs = with python3.pkgs; [
numpy
faster-whisper
ctranslate2
tqdm
sounddevice
];
build-system = [ python3.pkgs.setuptools ];
passthru.updateScript = nix-update-script { };
dependencies = with python3.pkgs; [
ctranslate2
faster-whisper
numpy
pyannote-audio
sounddevice
tqdm
];
nativeCheckInputs = with python3.pkgs; [
nose2
];
checkPhase = ''
runHook preCheck
# Note: we are not running the `e2e-tests` because they require downloading models from the internet.
${python3.interpreter} -m nose2 -s tests
runHook postCheck
'';
passthru.updateScript = nix-update-script { };
meta = with lib; {
description = "Whisper command line client compatible with original OpenAI client based on CTranslate2";
homepage = "https://github.com/Softcatala/whisper-ctranslate2";