File size: 327 Bytes
d8f20d1 | 1 2 3 4 5 6 7 8 9 10 | from diffusion import DiffusionConfig
def test_article_schedule_has_ten_rates_and_resolves_the_canvas() -> None:
config = DiffusionConfig()
assert config.mask_probabilities == (1.0, 0.9, 0.8, 0.7, 0.6, 0.5, 0.4, 0.3, 0.2, 0.1)
assert config.masks_after_step(1) == 216
assert config.masks_after_step(10) == 0
|