mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-09 23:12:02 +00:00
fix or silent pyright issues
This commit is contained in:
parent
65f19d192f
commit
2b2b6740b7
|
@ -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
|
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
|
timesteps = arange(start=0, end=self.num_inference_steps, step=1, device=self.device) * step_ratio
|
||||||
return timesteps.flip(0)
|
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
|
raise NotImplementedError
|
||||||
|
|
|
@ -64,7 +64,7 @@ def test_ddim_diffusers():
|
||||||
|
|
||||||
|
|
||||||
def test_euler_diffusers():
|
def test_euler_diffusers():
|
||||||
from diffusers import EulerDiscreteScheduler
|
from diffusers import EulerDiscreteScheduler # type: ignore
|
||||||
|
|
||||||
manual_seed(0)
|
manual_seed(0)
|
||||||
diffusers_scheduler = EulerDiscreteScheduler(
|
diffusers_scheduler = EulerDiscreteScheduler(
|
||||||
|
|
Loading…
Reference in a new issue