Updated cuda version, added jaxtyping and python widgets

mkranzlein · Feb 18, 2024 · f48583e · f48583e
1 parent f467138
commit f48583e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/environment.yml b/environment.yml
@@ -3,7 +3,8 @@ channels:
   - nvidia # For pytorch's GPU dependencies
 dependencies:
   - huggingface::transformers
-  - jaxtyping
+  - ipywidgets
+  - jsonlines
   - jupyterlab
   - networkx
   - numpy
@@ -13,8 +14,10 @@ dependencies:
   - pytest
   - pytest-cov
   - pytorch::pytorch
-  - pytorch::pytorch-cuda=11.7
+  - pytorch::pytorch-cuda=12.1
   - scikit-learn
+  - tqdm
   - pip:
+    - jaxtyping 
     - torcheval
     - typeguard