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

test_schedulers: fix pyright error

Browse source 
Due to changes in diffusers 0.25.0
This commit is contained in:
Cédric Deltheil 2024-01-10 16:34:59 +01:00 committed by Cédric Deltheil
parent 6dbaec3e56
commit ce0f9887a3
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -82,7 +82,9 @@ def test_euler_diffusers():
sample = randn(1, 4, 32, 32)
noise = randn(1, 4, 32, 32)
assert isclose(diffusers_scheduler.init_noise_sigma, refiners_scheduler.init_noise_sigma), "init_noise_sigma differ"
ref_init_noise_sigma = diffusers_scheduler.init_noise_sigma # type: ignore
assert isinstance(ref_init_noise_sigma, Tensor)
assert isclose(ref_init_noise_sigma, refiners_scheduler.init_noise_sigma), "init_noise_sigma differ"
for step, timestep in enumerate(diffusers_scheduler.timesteps):
diffusers_output = cast(Tensor, diffusers_scheduler.step(noise, timestep, sample).prev_sample) # type: ignore