dev/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Merge pull request #257002 from mweinelt/ctranslate-tensorflow-bin

Browse Source 
ctranslate2: 3.18.0 -> 3.20.0; python310Packages.ctranslate2: unblock by using tensorflow-bin
This commit is contained in:
Martin Weinelt 2023-09-24 20:07:01 +02:00 committed by GitHub
commit 2b9a9254e3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
pkgs/development/libraries/ctranslate2/default.nix
View File

@ -10,6 +10,10 @@
, withOneDNN ? false, oneDNN , withOneDNN ? false, oneDNN
, withOpenblas ? true, openblas , withOpenblas ? true, openblas
, withRuy ? true , withRuy ? true
# passthru tests
, libretranslate
, wyoming-faster-whisper
}: }:
let let
@ -17,13 +21,13 @@ let
in in
stdenv.mkDerivation rec { stdenv.mkDerivation rec {
pname = "ctranslate2"; pname = "ctranslate2";
version = "3.18.0"; version = "3.20.0";
src = fetchFromGitHub { src = fetchFromGitHub {
owner = "OpenNMT"; owner = "OpenNMT";
repo = "CTranslate2"; repo = "CTranslate2";
rev = "v${version}"; rev = "v${version}";
hash = "sha256-ipCUiCyWubKTUB0jDOsRN+DSg3S84hbj8Xum/2NsrKc="; hash = "sha256-PdCjzLyc5O1rrTtPz8JD08unY7uMNS5fcD3ZLHJDeYg=";
fetchSubmodules = true; fetchSubmodules = true;
}; };
@ -57,6 +61,13 @@ stdenv.mkDerivation rec {
darwin.apple_sdk.frameworks.CoreVideo darwin.apple_sdk.frameworks.CoreVideo
]; ];
passthru.tests = {
inherit
libretranslate
wyoming-faster-whisper
;
};
meta = with lib; { meta = with lib; {
description = "Fast inference engine for Transformer models"; description = "Fast inference engine for Transformer models";
homepage = "https://github.com/OpenNMT/CTranslate2"; homepage = "https://github.com/OpenNMT/CTranslate2";

4
pkgs/development/python-modules/ctranslate2/default.nix
View File

@ -12,7 +12,7 @@
# tests # tests
, pytestCheckHook , pytestCheckHook
, tensorflow , tensorflow-bin
, torch , torch
, transformers , transformers
, wurlitzer , wurlitzer
@ -49,7 +49,7 @@ buildPythonPackage rec {
nativeCheckInputs = [ nativeCheckInputs = [
pytestCheckHook pytestCheckHook
tensorflow tensorflow-bin
torch torch
transformers transformers
wurlitzer wurlitzer