Fix bug in requirements

eigenvivek · Nov 30, 2023 · f6e536e · f6e536e
1 parent fc37c4a
commit f6e536e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/environment.yml b/environment.yml
@@ -14,7 +14,7 @@ dependencies:
     - h5py
     - scikit-image
     - seaborn
-    - transformers
+    - pytorch-transformers
     - timm
     - torchmetrics
     - tqdm

diff --git a/settings.ini b/settings.ini
@@ -39,6 +39,6 @@ status = 3
 user = eigenvivek
 
 ### Optional ###
-requirements = diffdrr h5py scikit-image seaborn torch torchvision pytorch3d timm transformers torchmetrics tqdm beartype jaxtyping
+requirements = diffdrr h5py scikit-image seaborn torch torchvision pytorch3d timm pytorch-transformers torchmetrics tqdm beartype jaxtyping
 dev_requirements = jupyterlab_code_formatter black flake8 isort nbdev ipykernel jupyter-server-proxy 
 optional_requirements = submitit