From 37a74bd5491e890cff2e7ad2e34a90ed9d2f4ea6 Mon Sep 17 00:00:00 2001 From: limiteinductive Date: Sun, 3 Dec 2023 18:10:11 +0100 Subject: [PATCH] format test_scheduler file --- tests/foundationals/latent_diffusion/test_schedulers.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/tests/foundationals/latent_diffusion/test_schedulers.py b/tests/foundationals/latent_diffusion/test_schedulers.py index 845b672..01552f0 100644 --- a/tests/foundationals/latent_diffusion/test_schedulers.py +++ b/tests/foundationals/latent_diffusion/test_schedulers.py @@ -14,7 +14,13 @@ def test_scheduler_utils(): def _generate_timesteps(self) -> Tensor: return Tensor() - scheduler = DummyScheduler(num_inference_steps=10, num_train_timesteps=20, initial_diffusion_rate=0.1, final_diffusion_rate=0.2, device="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(