1 files changed, 1 insertions, 2 deletions

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]:
         """