From f581fe89cb0825adcf9a8172c558ffd2660fd1ea Mon Sep 17 00:00:00 2001 From: henk717 Date: Thu, 17 Mar 2022 21:11:36 +0100 Subject: [PATCH] Torch version changes --- environments/base.yml | 20 -------------------- environments/huggingface.yml | 2 +- environments/rocm.yml | 2 +- requirements.txt | 2 +- requirements_mtj.txt | 2 +- 5 files changed, 4 insertions(+), 24 deletions(-) delete mode 100644 environments/base.yml diff --git a/environments/base.yml b/environments/base.yml deleted file mode 100644 index d5a7df17..00000000 --- a/environments/base.yml +++ /dev/null @@ -1,20 +0,0 @@ -name: koboldai -channels: - - pytorch - - conda-forge - - defaults -dependencies: - - colorama - - flask-socketio - - pytorch - - python=3.8.* - - cudatoolkit=11.1 - - eventlet - - markdown - - bleach - - pip - - git - - pip: - - flask-cloudflared - - flask-ngrok - - lupa==1.10 diff --git a/environments/huggingface.yml b/environments/huggingface.yml index e0d559bc..f9380cf3 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -6,7 +6,7 @@ channels: dependencies: - colorama - flask-socketio - - pytorch=1.10.* + - pytorch=1.11.* - python=3.8.* - cudatoolkit=11.1 - eventlet diff --git a/environments/rocm.yml b/environments/rocm.yml index 30a17eb1..c1c19932 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -15,7 +15,7 @@ dependencies: - protobuf - pip: - --find-links https://download.pytorch.org/whl/rocm4.2/torch_stable.html - - torch==1.10.* + - torch==1.11.* - torchvision==0.11.1 - flask-cloudflared - flask-ngrok diff --git a/requirements.txt b/requirements.txt index 2563073c..9784986c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,7 +2,7 @@ transformers>=4.17 Flask Flask-SocketIO requests -torch>=1.9 +torch==1.11 flask-cloudflared flask-ngrok eventlet diff --git a/requirements_mtj.txt b/requirements_mtj.txt index 59cb5a5f..50d07023 100644 --- a/requirements_mtj.txt +++ b/requirements_mtj.txt @@ -1,4 +1,4 @@ -torch >= 1.9 +torch >= 1.9, <= 1.11 numpy tqdm requests