Dependency Fixes

environments/huggingface.yml

@@ -6,7 +6,7 @@ channels:
 dependencies:
   - colorama
   - flask-socketio
-  - pytorch
+  - pytorch=1.10.*
   - python=3.8.*
   - cudatoolkit=11.1
   - eventlet
@@ -20,4 +20,4 @@ dependencies:
     - flask-cloudflared
     - flask-ngrok
     - lupa==1.10
-    - git+https://github.com/huggingface/transformers
+    - transformers>=4.17

environments/rocm.yml

@@ -15,9 +15,9 @@ dependencies:
   - protobuf
   - pip:
     - --find-links https://download.pytorch.org/whl/rocm4.2/torch_stable.html
-    - torch
+    - torch==1.10.*
     - torchvision==0.11.1
     - flask-cloudflared
     - flask-ngrok
     - lupa==1.10
-    - git+https://github.com/huggingface/transformers
+    - transformers>=4.17

requirements.txt

@@ -1,8 +1,8 @@
-git+https://github.com/huggingface/transformers
+transformers>=4.17
 Flask
 Flask-SocketIO
 requests
-torch
+torch==1.10.*
 flask-cloudflared
 flask-ngrok
 eventlet

requirements_mtj.txt

@@ -5,7 +5,7 @@ optax >= 0.0.5, <= 0.0.9
 dm-haiku == 0.0.5
 ray[default]
 jax == 0.2.21
-transformers
+transformers >= 4.17
 progressbar2
 git+https://github.com/VE-FORBRYDERNE/mesh-transformer-jax@ck
 flask