fix conda environment

Signed-off-by: Vibhu Jawa <vibhujawa@gmail.com>
rapidsai · Oct 3, 2024 · ae7dd72 · ae7dd72
1 parent a57366a
commit ae7dd72
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