Un-pegging Transformer version

environments/huggingface.yml

@@ -27,7 +27,7 @@ dependencies:
     - flask-cloudflared
     - flask-ngrok
     - lupa==1.10
-    - transformers==4.21.3
+    - transformers>=4.20.1
     - accelerate
     - flask-session
     - python-socketio[client]

requirements.txt

@@ -1,4 +1,4 @@
-transformers==4.21.3
+transformers>=4.20.1
 Flask
 Flask-SocketIO
 requests

requirements_mtj.txt

@@ -5,7 +5,7 @@ requests
 dm-haiku == 0.0.5
 jax == 0.2.21
 jaxlib >= 0.1.69, <= 0.3.7
-transformers ==4.21.3
+transformers >=4.20.1
 progressbar2
 git+https://github.com/VE-FORBRYDERNE/mesh-transformer-jax@ck
 flask

utils.py

@@ -188,7 +188,7 @@ def _download_with_aria2(aria2_config: str, total_length: int, directory: str =
                 try:
                     print('\r' + bar, end='')
                     try:
-                        emit('from_server', {'cmd': 'model_load_status', 'data': bar.replace(" ", " ")}, broadcast=True)
+                        emit('from_server', {'cmd': 'model_load_status', 'data': bar.replace(" ", " ")}, broadcast=True, room="UI_1")
                     except:
                         pass
                     eventlet.sleep(seconds=0)