From 6c072fe50b3bfc561f22e5d591212d30de3c2dd2 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Mon, 3 Oct 2022 12:08:16 +0200 Subject: Fixed euler_a generator argument --- schedulers/scheduling_euler_a.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) (limited to 'schedulers') diff --git a/schedulers/scheduling_euler_a.py b/schedulers/scheduling_euler_a.py index d7fea85..c6436d8 100644 --- a/schedulers/scheduling_euler_a.py +++ b/schedulers/scheduling_euler_a.py @@ -198,7 +198,7 @@ class EulerAScheduler(SchedulerMixin, ConfigMixin): timestep: int, timestep_prev: int, sample: torch.FloatTensor, - generator: None, + generator: torch.Generator = None, return_dict: bool = True, ) -> Union[SchedulerOutput, Tuple]: """ @@ -240,7 +240,6 @@ class EulerAScheduler(SchedulerMixin, ConfigMixin): sample_hat: torch.FloatTensor, sample_prev: torch.FloatTensor, derivative: torch.FloatTensor, - generator: None, return_dict: bool = True, ) -> Union[SchedulerOutput, Tuple]: """ -- cgit v1.2.3-54-g00ecf