add to reqirements

model.py

@@ -3,6 +3,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import warnings
 
 
+
+
+
 torch_dtype = torch.bfloat16
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model_name = "mosaicml/mpt-7b"

requirements.txt

@@ -4,4 +4,6 @@ sentencepiece==0.1.*
 torch==2.*
 transformers==4.*
 uvicorn[standard]==0.17.*
-streamlit-ace
+streamlit-ace
+einops
+langchain