xavierk
80a3792576
Import the vocalinux source tree (tray app, Next.js site in web/, packaging, installers, docs and tests) on top of the Gitea template commit. The project README replaces the template README, and the template CI files (.gitea/workflows/ci.yml, .jscpd.json, CI.md) are kept.
110 lines
3.3 KiB
Nix
110 lines
3.3 KiB
Nix
# pywhispercpp is not in nixpkgs (same gap the .deb/.rpm note in
|
|
# packaging/native/nfpm.yaml calls out for Debian and Fedora), so the flake
|
|
# builds it here. The PyPI sdist already vendors whisper.cpp and pybind11, so
|
|
# no extra source fetches are needed; setup.py drives CMake from build_ext.
|
|
{
|
|
lib,
|
|
buildPythonPackage,
|
|
fetchPypi,
|
|
setuptools,
|
|
wheel,
|
|
cmake,
|
|
shaderc,
|
|
spirv-headers,
|
|
vulkan-headers,
|
|
vulkan-loader,
|
|
numpy,
|
|
requests,
|
|
tqdm,
|
|
platformdirs,
|
|
autoPatchelfHook,
|
|
stdenv,
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "pywhispercpp";
|
|
version = "1.5.1";
|
|
pyproject = true;
|
|
|
|
src = fetchPypi {
|
|
inherit pname version;
|
|
hash = "sha256-XfiX5d2dnxaAT8k3vYW59YgKCymlX4hDWF4OA9dhleU=";
|
|
};
|
|
|
|
# python -m build verifies every build-system.requires entry resolves to an
|
|
# installed Python package. The ones setup.py never imports are dropped:
|
|
# ninja is try/except'd (Unix Makefiles are the fallback), repairwheel only
|
|
# runs on Windows, setuptools-scm is unused, and cmake is a PATH tool, not a
|
|
# Python package.
|
|
postPatch = ''
|
|
substituteInPlace pyproject.toml \
|
|
--replace-fail ' "ninja",' "" \
|
|
--replace-fail ' "cmake>=3.12",' "" \
|
|
--replace-fail ' "repairwheel",' "" \
|
|
--replace-fail ' "setuptools-scm>=8"' ""
|
|
|
|
# setup() passes no version, so without setuptools-scm the wheel metadata
|
|
# lands as 0.0.0; get_version() reads PYWHISPERCPP_VERSION instead.
|
|
substituteInPlace setup.py \
|
|
--replace-fail ' name="pywhispercpp",' ' name="pywhispercpp", version=get_version(),'
|
|
'';
|
|
|
|
env = {
|
|
# The sdist ships no version.txt; setup.py falls back to this env var.
|
|
PYWHISPERCPP_VERSION = version;
|
|
# Build Vulkan as a shared backend: Vocalinux's GPU detector looks for
|
|
# libggml-vulkan.so. CMake's build-tree RPATH can leak /build into the
|
|
# wheel; disable CMake RPATHs and let autoPatchelfHook set Nix runtime paths.
|
|
CMAKE_ARGS = "-DBUILD_SHARED_LIBS=ON -DGGML_NATIVE=OFF -DGGML_VULKAN=ON -DCMAKE_POSITION_INDEPENDENT_CODE=ON -DCMAKE_SKIP_RPATH=ON";
|
|
};
|
|
|
|
# env attrs are exported verbatim; expanding the core count needs a shell
|
|
# phase.
|
|
preBuild = ''
|
|
export CMAKE_BUILD_PARALLEL_LEVEL="$NIX_BUILD_CORES"
|
|
'';
|
|
|
|
# cmake is only here so setup.py's build_ext can invoke it; letting the
|
|
# cmake configure hook run standalone leaves a build/ dir that breaks the
|
|
# pypa wheel build.
|
|
dontUseCmakeConfigure = true;
|
|
|
|
nativeBuildInputs = [
|
|
cmake
|
|
shaderc # provides glslc, used to compile whisper.cpp's Vulkan shaders
|
|
autoPatchelfHook
|
|
];
|
|
|
|
build-system = [
|
|
setuptools
|
|
wheel
|
|
];
|
|
|
|
# CMake's Vulkan backend calls find_package(SPIRV-Headers CONFIG REQUIRED),
|
|
# so expose the package configuration as well as the Vulkan headers/loader.
|
|
buildInputs = [
|
|
stdenv.cc.cc.lib
|
|
spirv-headers
|
|
vulkan-headers
|
|
vulkan-loader
|
|
];
|
|
|
|
dependencies = [
|
|
numpy
|
|
requests
|
|
tqdm
|
|
platformdirs
|
|
];
|
|
|
|
# pywhispercpp.model pulls in the compiled _pywhispercpp extension, so
|
|
# checking all three proves the native binding loads, not just the package.
|
|
pythonImportsCheck = [ "pywhispercpp" "pywhispercpp.model" "_pywhispercpp" ];
|
|
|
|
meta = {
|
|
description = "Python bindings for whisper.cpp";
|
|
homepage = "https://github.com/absadiki/pywhispercpp";
|
|
license = lib.licenses.mit;
|
|
platforms = lib.platforms.linux;
|
|
};
|
|
}
|