diff options
Diffstat (limited to 'PKGBUILD')
-rw-r--r-- | PKGBUILD | 59 |
1 files changed, 39 insertions, 20 deletions
@@ -1,29 +1,48 @@ -# Maintainer: Filip Graliński <filipg@amu.edu.pl> +# Maintainer: Daniel Bershatsky <bepshatsky@yandex.ru> +# Contributor: Filip Graliński <filipg@amu.edu.pl> pkgname=python-transformers -pkgver=4.11.3 -pkgrel=2 +_pkgname=${pkgname#python-} +pkgver=4.40.1 +pkgrel=1 pkgdesc="State-of-the-art Natural Language Processing for Jax, PyTorch and TensorFlow" -arch=('i686' 'x86_64') -url="https://pypi.org/project/transformers" -license=('Apache License 2.0') -depends=('cuda' - 'nccl' - 'python-filelock' - 'python-pytorch' - 'python-sacremoses' - 'python-tokenizers' - 'python-tqdm') - -source=("https://files.pythonhosted.org/packages/52/9a/d7a8dea8ebab15198877ed5b6d718ed04373987854a21e91c1948d555456/transformers-${pkgver}.tar.gz") -sha256sums=('755b052df58906f122f7166c573c22531416eab8a9f59c44ff7148be12e62621') +arch=('any') +url='https://github.com/huggingface/transformers' +license=('Apache') +depends=( + 'python-filelock' + 'python-huggingface-hub' + 'python-numpy' + 'python-packaging' + 'python-regex' + 'python-requests' + 'python-safetensors' + 'python-tokenizers>=0.19' + 'python-tqdm' + 'python-yaml' +) +makedepends=('python-build' 'python-installer' 'python-setuptools' 'python-wheel') +optdepends=( + 'python-bitsandbytes: 8-bit support for PyTorch' + 'python-flax: JAX support' + 'python-onnxconverter-common: TensorFlow support' + 'python-pytorch: PyTorch support' + 'python-tensorflow: TensorFlow support' + 'python-tf2onnx: TensorFlow support' +) +source=( + "python-transformers-$pkgver.tar.gz"::"https://github.com/huggingface/transformers/archive/refs/tags/v$pkgver.tar.gz" +) +sha256sums=('99467420e37a174c2a0855ca1a3451904041b2894e56f41f449522a72c6af13c') build() { - cd transformers-$pkgver - python setup.py build + python -m build -nw "transformers-$pkgver" } package() { - cd transformers-$pkgver - python setup.py install --root="$pkgdir" --optimize=1 + cd "transformers-$pkgver" + python -m installer \ + --compile-bytecode 1 \ + --destdir $pkgdir \ + $srcdir/$_pkgname-$pkgver/dist/transformers-$pkgver-*-*.whl } |