diff --git a/environments/huggingface.yml b/environments/huggingface.yml index 56b97938..e0d559bc 100644 --- a/environments/huggingface.yml +++ b/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 \ No newline at end of file + - transformers>=4.17 \ No newline at end of file diff --git a/environments/rocm.yml b/environments/rocm.yml index b8e74515..30a17eb1 100644 --- a/environments/rocm.yml +++ b/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 \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index 9c1640e5..8b1b36a5 100644 --- a/requirements.txt +++ b/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 diff --git a/requirements_mtj.txt b/requirements_mtj.txt index ad205326..9805328f 100644 --- a/requirements_mtj.txt +++ b/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