geoffrey/dotfiles
geoffrey/dotfiles
1
0
Fork 0
Code Issues 7 Pull requests Releases Activity

Upgrade whisperx

Browse source
This commit is contained in:
Geoffrey Frogeye 2024-12-20 23:24:55 +01:00
parent a5be7cb695
commit b61685f677
Signed by: geoffrey
GPG key ID: C72403E7F82E6AD8
3 changed files with 11 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

14
hm/extra/whisperx.nix
View file

@ -1,15 +1,15 @@
{
pkgs ? import <nixpkgs> { },
}:
pkgs.python3Packages.buildPythonPackage {
pkgs.python3Packages.buildPythonPackage rec {
pname = "whisperx";
version = "2024-08-19";
version = "3.2.0";
# pypi doesn't have the requirements.txt file, and it's required
src = pkgs.fetchFromGitHub {
owner = "m-bain";
repo = "whisperX";
rev = "9e3a9e0e38fcec1304e1784381059a0e2c670be5"; # git doesn't have tags
hash = "sha256-IVtn9fe/yi4+fbH57s9LoiREnMZ2nhEObp1a4R/7gHg=";
rev = "v${version}"; # git doesn't have tags
hash = "sha256-JQvyR9JW8OiSRI0eywTyWB4VMXmu6cTpBekBWonoJa4=";
};
pyproject = true;
dependencies = [
@ -29,6 +29,12 @@ pkgs.python3Packages.buildPythonPackage {
pkgs.python3Packages.pandas
pkgs.python3Packages.nltk
];
pythonRelaxDeps = true;
# torchaudio: 2.5.1a0 is >=2, despite dependency check saying otherwise
# ctranslate2: was pinned to 4.4.0 to fix some nvidia pinning issue or whatnot
# pyannote-audio: not sure what pins this to 3.1.1, but 3.3.1 works fine
# For some reason pyannote-audio doesn't get relaxed when listed in a pythonRelaxDeps array,
# hence why using true
build-system = [
pkgs.python3Packages.setuptools
];