Torch version changes

environments/base.yml

@@ -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

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

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

requirements.txt

@@ -2,7 +2,7 @@ transformers>=4.17
 Flask
 Flask-SocketIO
 requests
-torch>=1.9
+torch==1.11
 flask-cloudflared
 flask-ngrok
 eventlet

requirements_mtj.txt

@@ -1,4 +1,4 @@
-torch >= 1.9
+torch >= 1.9, <= 1.11
 numpy
 tqdm
 requests