From ca74884bd710c271923203ca4a154f67a55bf3be Mon Sep 17 00:00:00 2001
From: Weves <chrisweaver101@gmail.com>
Date: Sun, 8 Oct 2023 16:09:36 -0700
Subject: [PATCH] Pin pytorch version to fix segmentation fault in Docker

---
 backend/requirements/default.txt | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/backend/requirements/default.txt b/backend/requirements/default.txt
index 9df67ed5a..11fee09b0 100644
--- a/backend/requirements/default.txt
+++ b/backend/requirements/default.txt
@@ -49,6 +49,8 @@ slack-sdk==3.20.2
 SQLAlchemy[mypy]==2.0.15
 tensorflow==2.13.0
 tiktoken==0.4.0
+torch==2.0.1
+torchvision==0.15.2
 transformers==4.30.1
 typesense==0.15.1
 uvicorn==0.21.1