env fix

allenai · Sep 23, 2024 · 491b738 · 491b738
1 parent 1cf3cd8
commit 491b738
Showing 1 changed file with 11 additions and 15 deletions.
diff --git a/gantry-environment.yaml b/gantry-environment.yaml
@@ -37,22 +37,18 @@ dependencies:
   - sphinx-copybutton==0.5.2
   - sphinx-autobuild==2021.3.14
   - sphinx-autodoc-typehints==1.23.3
-
-  # CUDA 11.8 specific installation for PyTorch
   - cudatoolkit=11.8
 
-  # Additional dependencies to be installed via pip
-  pip:
-    - pymupdf
-    - pypdf
-    - pypdfium2
-    - transformers @ git+https://github.com/huggingface/transformers.git@f9b44097261270377199bb0fcbceb01fd9f7d0c0
-    - lingua-language-detector
-    - necessary
-
-  # CUDA PyTorch package with index URL
+  # pip section for additional packages
   - pip:
-    - --index-url https://download.pytorch.org/whl/cu118
-    - torch
-    - torchvision
+      - pymupdf
+      - pypdf
+      - pypdfium2
+      - transformers @ git+https://github.com/huggingface/transformers.git@f9b44097261270377199bb0fcbceb01fd9f7d0c0
+      - lingua-language-detector
+      - necessary
 
+      # CUDA PyTorch package with index URL
+      - --index-url https://download.pytorch.org/whl/cu118
+      - torch
+      - torchvision