Merge branch 'rapidsai:main' into fix_torch_logging

rapidsai · Oct 9, 2024 · 1ba728e · 1ba728e
2 parents efbc828 + ba49e35
commit 1ba728e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/conda/environments/cuda_dev.yaml b/conda/environments/cuda_dev.yaml
@@ -20,13 +20,15 @@ dependencies:
   - pytest
   - pytest-benchmark
   - pytest-cov>=2
-  - pytorch>=2.0
+  - pytorch>=2.0=*cuda*
+  - tiktoken
   - transformers
   - tokenizers
   - torchvision
   - scikit-learn>=0.23.1
   - scipy
   - setuptools>=61.0.0
+  - sentencepiece
   - sentence-transformers
   - pip:
     - pytrec-eval
diff --git a/requirements/cuda12x.txt b/requirements/cuda12x.txt
@@ -11,3 +11,4 @@ curated-transformers>=1.0
 bitsandbytes>=0.30
 sentence-transformers>=2.0
 sentencepiece
+tiktoken