natsukium 2023-07-27 22:58:48 +09:00 committed by Yt
parent 5b01208470
commit 2b63ca9fb8

View file

@ -51,7 +51,7 @@
buildPythonPackage rec { buildPythonPackage rec {
pname = "transformers"; pname = "transformers";
version = "4.30.2"; version = "4.31.0";
format = "setuptools"; format = "setuptools";
disabled = pythonOlder "3.8"; disabled = pythonOlder "3.8";
@ -60,14 +60,13 @@ buildPythonPackage rec {
owner = "huggingface"; owner = "huggingface";
repo = pname; repo = pname;
rev = "refs/tags/v${version}"; rev = "refs/tags/v${version}";
hash = "sha256-S1jQsBObKGZY9tlbcNcgchwUs/eeaohYxOtbN1cPa2Q="; hash = "sha256-YbLI/CkRto8G4bV7ijUkB/0cc7LkfNBQxL1iNv8aWW4=";
}; };
propagatedBuildInputs = [ propagatedBuildInputs = [
filelock filelock
huggingface-hub huggingface-hub
numpy numpy
protobuf
packaging packaging
pyyaml pyyaml
regex regex
@ -91,8 +90,11 @@ buildPythonPackage rec {
ja = [ ja = [
# fugashi # fugashi
# ipadic # ipadic
# unidic_lite # rhoknp
# sudachidict_core
# sudachipy
# unidic # unidic
# unidic_lite
]; ];
sklearn = [ sklearn = [
scikit-learn scikit-learn
@ -122,6 +124,7 @@ buildPythonPackage rec {
onnxconverter-common onnxconverter-common
tf2onnx tf2onnx
onnxruntime onnxruntime
onnxruntime-tools
]; ];
modelcreation = [ modelcreation = [
cookiecutter cookiecutter