From ea738180de17b88dfd155cb25197ff65e2c9c6af Mon Sep 17 00:00:00 2001 From: ebolam Date: Fri, 7 Oct 2022 15:18:40 -0400 Subject: [PATCH] Un-pegging Transformer version --- environments/huggingface.yml | 2 +- requirements.txt | 2 +- requirements_mtj.txt | 2 +- utils.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/environments/huggingface.yml b/environments/huggingface.yml index 4521a783..0c8d71dc 100644 --- a/environments/huggingface.yml +++ b/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] diff --git a/requirements.txt b/requirements.txt index fa2504cf..b068764e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -transformers==4.21.3 +transformers>=4.20.1 Flask Flask-SocketIO requests diff --git a/requirements_mtj.txt b/requirements_mtj.txt index 73fac413..f2798d32 100644 --- a/requirements_mtj.txt +++ b/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 diff --git a/utils.py b/utils.py index c6250ef6..15dd9307 100644 --- a/utils.py +++ b/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)