Add a few requirements for ensured parity with turbine-models require…

…ments. (#2142) * Add scipy to requirements. Adds diffusers req and a note for torchsde.
nod-ai · May 28, 2024 · 151195a · 151195a
1 parent 8146f0b
commit 151195a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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