Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-09-19 19:05:28 +00:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

fix or silent pyright issues

Browse source
This commit is contained in:
Cédric Deltheil 2024-01-10 11:45:19 +01:00 committed by Cédric Deltheil
parent 65f19d192f
commit 2b2b6740b7
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/refiners/foundationals/latent_diffusion/schedulers/ddpm.py
View file

@ -1,4 +1,4 @@
from torch import Tensor, arange, device as Device
from torch import Generator, Tensor, arange, device as Device
from refiners.foundationals.latent_diffusion.schedulers.scheduler import Scheduler
@ -30,5 +30,5 @@ class DDPM(Scheduler):
timesteps = arange(start=0, end=self.num_inference_steps, step=1, device=self.device) * step_ratio
return timesteps.flip(0)
def __call__(self, x: Tensor, noise: Tensor, step: int) -> Tensor:
def __call__(self, x: Tensor, noise: Tensor, step: int, generator: Generator | None = None) -> Tensor:
raise NotImplementedError

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

@ -64,7 +64,7 @@ def test_ddim_diffusers():
def test_euler_diffusers():
from diffusers import EulerDiscreteScheduler
from diffusers import EulerDiscreteScheduler # type: ignore
manual_seed(0)
diffusers_scheduler = EulerDiscreteScheduler(