diff --git a/pyproject.toml b/pyproject.toml
index b62bf65d..b3afd523 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -72,6 +72,7 @@ qdrant-client = "*"
 sentence-transformers = "*"
 peft = "*"
 modelscope = "1.10.0"
+psutil = "*"
 
 
 [tool.poetry.group.lint.dependencies]
diff --git a/requirements.txt b/requirements.txt
index 05037793..27da7c12 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -59,4 +59,5 @@ mkdocs-material
 mkdocs-glightbox
 pre-commit==3.2.2
 peft
-modelscope
\ No newline at end of file
+modelscope
+psutil
\ No newline at end of file