mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-09 23:12:02 +00:00
fix test_scheduler_utils
This commit is contained in:
parent
1075ea4a62
commit
6f110ee2b2
|
@ -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(
|
||||
|
|
Loading…
Reference in a new issue