summarylogtreecommitdiffstats
path: root/PKGBUILD
blob: 894b9ff60f00618032317873230ec0feecbb20fd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
# Maintainer: Filip GraliƄski <filipg@amu.edu.pl>

pkgname=python-transformers
pkgver=4.11.3
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-pytorch')
source=("https://files.pythonhosted.org/packages/52/9a/d7a8dea8ebab15198877ed5b6d718ed04373987854a21e91c1948d555456/transformers-${pkgver}.tar.gz")
sha256sums=('755b052df58906f122f7166c573c22531416eab8a9f59c44ff7148be12e62621')

build() {
    cd transformers-$pkgver
    python setup.py build
}

package() {
    cd transformers-$pkgver
    python setup.py install --root="$pkgdir" --optimize=1
}