diff --git a/pyproject.toml b/pyproject.toml
index 8707c6c911..03396312ac 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -83,6 +83,7 @@ dependencies = [
   "torch~=2.0.0",
   "torchvision>=0.14.1",
   "torchmetrics",
+  "torchsde==0.2.5",
   "transformers~=4.30",
   "uvicorn[standard]==0.21.1",
   "windows-curses; sys_platform=='win32'",