diff --git a/environments/huggingface.yml b/environments/huggingface.yml index 5942d79a..1bef463d 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -5,12 +5,13 @@ channels: - defaults dependencies: - colorama - - flask-socketio - - flask-session + - flask-socketio=5.3.2 + - flask-session=0.4.0 + - python-socketio=5.7.2 - pytorch=1.11.* - python=3.8.* - cudatoolkit=11.1 - - eventlet + - eventlet=0.33.3 - dnspython=2.2.1 - markdown - bleach=4.1.0 diff --git a/environments/rocm.yml b/environments/rocm.yml index c9c6b487..47d29e07 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -4,10 +4,11 @@ channels: - defaults dependencies: - colorama - - flask-socketio - - flask-session + - flask-socketio=5.3.2 + - flask-session=0.4.0 + - python-socketio=5.7.2 - python=3.8.* - - eventlet + - eventlet=0.33.3 - dnspython=2.2.1 - markdown - bleach=4.1.0 diff --git a/requirements.txt b/requirements.txt index b79ca5cb..b13230ed 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,12 +1,13 @@ transformers==4.24.0 huggingface_hub==0.12.1 -Flask -Flask-SocketIO +Flask==2.2.3 +Flask-SocketIO==5.3.2 +python-socketio==5.7.2 requests torch >= 1.9, < 1.13 flask-cloudflared==0.0.10 flask-ngrok -eventlet +eventlet==0.33.3 dnspython==2.2.1 lupa==1.10 markdown @@ -14,7 +15,7 @@ bleach==4.1.0 sentencepiece protobuf accelerate -flask-session +flask-session==0.4.0 marshmallow>=3.13 apispec-webframeworks loguru diff --git a/requirements_mtj.txt b/requirements_mtj.txt index c99852e7..e66e1ba7 100644 --- a/requirements_mtj.txt +++ b/requirements_mtj.txt @@ -9,16 +9,17 @@ transformers == 4.24.0 huggingface_hub==0.12.1 progressbar2 git+https://github.com/VE-FORBRYDERNE/mesh-transformer-jax@ck -flask -Flask-SocketIO +Flask==2.2.3 +Flask-SocketIO==5.3.2 +python-socketio==5.7.2 flask-cloudflared==0.0.10 flask-ngrok -eventlet +eventlet==0.33.3 dnspython==2.2.1 lupa==1.10 markdown bleach==4.1.0 -flask-session +flask-session==0.4.0 marshmallow>=3.13 apispec-webframeworks loguru