Add a few requirements for ensured parity with turbine-models requirements. (#2142)

* Add scipy to requirements. Adds diffusers req and a note for torchsde.
2026-01-08 21:38:04 -05:00 · 2024-05-28 15:37:31 -05:00
parent 8146f0bd2f
commit 151195ab74
1 changed files with 3 additions and 0 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -24,6 +24,9 @@ parameterized
 # Add transformers, diffusers and scipy since it most commonly used
 #accelerate is now required for diffusers import from ckpt.
 accelerate
+scipy
+transformers==4.37.1
+torchsde # Required for Stable Diffusion SDE schedulers.
 ftfy
 gradio==4.29.0
 altair