fix: memory error

main.py

@@ -4,6 +4,10 @@ import shutil
 import subprocess
 import torch
 import json
+import faulthandler
+
+faulthandler.enable()
+
 
 from fastapi import FastAPI, HTTPException, UploadFile, WebSocket, WebSocketDisconnect
 from fastapi.staticfiles import StaticFiles
@@ -123,8 +127,7 @@ def run_ingest_route():
             retriever=RETRIEVER,
             return_source_documents=SHOW_SOURCES,
             chain_type_kwargs={
-                "prompt": prompt,
-                "memory": memory
+                "prompt": prompt
             },
         )
 

requirements.txt

@@ -31,6 +31,7 @@ websockets
 pydantic
 redis
 aioredis
+faulthandler
 
 # Streamlit related
 streamlit