fix test_scheduler_utils

This commit is contained in:
limiteinductive 2023-12-03 17:49:28 +01:00 committed by Benjamin Trom
parent 1075ea4a62
commit 6f110ee2b2

View file

@ -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(