Apply patch by Chris M.

1 files changed, 17 insertions, 5 deletions

diff --git a/PKGBUILD b/PKGBUILD
index 4763dde9e383..28a13e8e582a 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -7,13 +7,25 @@ pkgdesc="State-of-the-art Natural Language Processing for Jax, PyTorch and Tenso
 arch=('i686' 'x86_64')
 url="https://pypi.org/project/transformers"
 license=('Apache License 2.0')
-depends=('cuda'
-         'nccl'
-         'python-filelock'
-         'python-pytorch'
-         'python-sacremoses'
+depends=('python-filelock'
          'python-tokenizers'
+         'python-huggingface-hub'
+         'python-numpy'
+         'python-packaging'
+         'python-yaml'
+         'python-regex'
+         'python-requests'
          'python-tqdm')
+optdepends=(
+       'python-pytorch: PyTorch support'
+       'python-tensorflow: TensorFlow support'
+       'python-onnxconverter-common: TensorFlow support'
+       'python-tf2onn: TensorFlow support'
+       'python-jax: JAX support'
+       'python-jaxlib: JAX support'
+       'python-flax: JAX support'
+       'python-optax: JAX support'
+)
 
 source=("https://github.com/huggingface/transformers/archive/refs/tags/v${pkgver}.tar.gz")
 sha256sums=('a1cdffb59b0a409cb5de414fcfaf5208f4526023cd021245f37f309bb15673a9')