mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-21 21:58:47 +00:00
format test_scheduler file
This commit is contained in:
parent
90db6ef59d
commit
37a74bd549
|
@ -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(
|
||||
|
|
Loading…
Reference in a new issue