From 6f110ee2b234a8272ec6646065061f7d22ffa0e9 Mon Sep 17 00:00:00 2001 From: limiteinductive Date: Sun, 3 Dec 2023 17:49:28 +0100 Subject: [PATCH] fix test_scheduler_utils --- tests/foundationals/latent_diffusion/test_schedulers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/foundationals/latent_diffusion/test_schedulers.py b/tests/foundationals/latent_diffusion/test_schedulers.py index 391c109..845b672 100644 --- a/tests/foundationals/latent_diffusion/test_schedulers.py +++ b/tests/foundationals/latent_diffusion/test_schedulers.py @@ -14,7 +14,7 @@ def test_scheduler_utils(): def _generate_timesteps(self) -> Tensor: return Tensor() - scheduler = DummyScheduler(10, 20, 0.1, 0.2, "cpu") + scheduler = DummyScheduler(num_inference_steps=10, num_train_timesteps=20, initial_diffusion_rate=0.1, final_diffusion_rate=0.2, device="cpu") scale_factors = ( 1.0 - linspace(