mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-24 23:28:45 +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:
|
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(
|
||||||
|
|
Loading…
Reference in a new issue