chore: Add torchsde as a dependency for the SDE schedulers

This commit is contained in:
blessedcoolant 2023-06-19 23:20:53 +12:00
parent 9d4b84ef68
commit 0ae6325353

View File

@ -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'",