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