Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-09 23:12:02 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

format test_scheduler file

Browse source
This commit is contained in:
limiteinductive 2023-12-03 18:10:11 +01:00 committed by Benjamin Trom
parent 90db6ef59d
commit 37a74bd549
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/foundationals/latent_diffusion/test_schedulers.py
View file

@ -14,7 +14,13 @@ def test_scheduler_utils():
def _generate_timesteps(self) -> Tensor: def _generate_timesteps(self) -> Tensor:
return 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 = ( scale_factors = (
1.0 1.0
- linspace( - linspace(