From 7380518f246684e14b988485a2f8b9f55657707d Mon Sep 17 00:00:00 2001
From: Debanjum Singh Solanky <debanjum@gmail.com>
Date: Thu, 5 Jan 2023 15:09:06 -0300
Subject: [PATCH] Upgrade PyTorch, Pillow version to resolve Dependabot
 Security Advisories

This also enables GPU usage by Khoj on MacOS as MPS support is now in
PyTorch mainline
---
 config/environment.yml | 11 ++++++-----
 setup.py               |  6 +++---
 2 files changed, 9 insertions(+), 8 deletions(-)

diff --git a/config/environment.yml b/config/environment.yml
index a21b692c..01a37d8b 100644
--- a/config/environment.yml
+++ b/config/environment.yml
@@ -4,18 +4,19 @@ channels:
 dependencies:
   - python=3.8.*
   - numpy=1.22.4
-  - pytorch=1.11.0
-  - transformers=4.19.4
+  - pytorch=1.13.1
+  - torchvision=0.14.1
+  - transformers=4.21.0
   - sentence-transformers=2.1.0
   - fastapi=0.77.1
   - uvicorn=0.17.6
   - pyyaml=6.0
   - pytest=7.1.2
-  - pillow=8.4.0
-  - torchvision=0.12.0
+  - pillow=9.3.0
   - openai=0.20.0
   - pydantic=1.9.1
   - jinja2=3.1.2
   - aiofiles=0.8.0
   - huggingface_hub=0.8.1
-  - dateparser=1.1.1
\ No newline at end of file
+  - dateparser=1.1.1
+  - schedule=1.1.0
\ No newline at end of file
diff --git a/setup.py b/setup.py
index db8eee10..6c0c5033 100644
--- a/setup.py
+++ b/setup.py
@@ -24,8 +24,8 @@ setup(
     ),
     install_requires=[
         "numpy == 1.22.4",
-        "torch == 1.12.1",
-        "torchvision == 0.13.1",
+        "torch == 1.13.1",
+        "torchvision == 0.14.1",
         "transformers == 4.21.0",
         "sentence-transformers == 2.1.0",
         "openai == 0.20.0",
@@ -36,7 +36,7 @@ setup(
         "jinja2 == 3.1.2",
         "pyyaml == 6.0",
         "pytest == 7.1.2",
-        "pillow == 9.2.0",
+        "pillow == 9.3.0",
         "aiofiles == 0.8.0",
         "dateparser == 1.1.1",
         "pyqt6 == 6.3.1",