Add transformers torch extras base requirements

NVIDIA-Merlin · May 12, 2023 · b760f64 · b760f64
1 parent a070e77
commit b760f64
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/requirements/base_external.txt b/requirements/base_external.txt
@@ -1,3 +1,4 @@
-transformers>=4.12,<5
+transformers[torch]>=4.12,<5
 tqdm>=4.27
-pyarrow>=1.0
+pyarrow>=1.0
+torchmetrics>=0.10.0