From 151195ab743e67b6ca8f52ea2b2756897b5bfc65 Mon Sep 17 00:00:00 2001
From: Ean Garvey <87458719+monorimet@users.noreply.github.com>
Date: Tue, 28 May 2024 15:37:31 -0500
Subject: [PATCH] Add a few requirements for ensured parity with turbine-models
 requirements. (#2142)

* Add scipy to requirements.

Adds diffusers req and a note for torchsde.
---
 requirements.txt | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/requirements.txt b/requirements.txt
index bbb8adf6b1..eb5c9b91bb 100644
--- 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