summaryrefslogtreecommitdiff
path: root/share/spack/gitlab/cloud_pipelines/stacks/ml-cuda/spack.yaml
diff options
context:
space:
mode:
Diffstat (limited to 'share/spack/gitlab/cloud_pipelines/stacks/ml-cuda/spack.yaml')
-rw-r--r--share/spack/gitlab/cloud_pipelines/stacks/ml-cuda/spack.yaml3
1 files changed, 3 insertions, 0 deletions
diff --git a/share/spack/gitlab/cloud_pipelines/stacks/ml-cuda/spack.yaml b/share/spack/gitlab/cloud_pipelines/stacks/ml-cuda/spack.yaml
index ada465fd1a..9d050b5b01 100644
--- a/share/spack/gitlab/cloud_pipelines/stacks/ml-cuda/spack.yaml
+++ b/share/spack/gitlab/cloud_pipelines/stacks/ml-cuda/spack.yaml
@@ -26,6 +26,9 @@ spack:
# Horovod
- py-horovod
+ # Hugging Face
+ - py-transformers
+
# JAX
# https://github.com/google/jax/issues/12614
# - py-jax