diff options
| author | Volpeon <git@volpeon.ink> | 2022-10-03 12:08:16 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2022-10-03 12:08:16 +0200 |
| commit | 6c072fe50b3bfc561f22e5d591212d30de3c2dd2 (patch) | |
| tree | e6dd60b5fa696d614ccc1cddb869c12c29f6ab46 /schedulers | |
| parent | Assign unused images in validation dataset to train dataset (diff) | |
| download | textual-inversion-diff-6c072fe50b3bfc561f22e5d591212d30de3c2dd2.tar.gz textual-inversion-diff-6c072fe50b3bfc561f22e5d591212d30de3c2dd2.tar.bz2 textual-inversion-diff-6c072fe50b3bfc561f22e5d591212d30de3c2dd2.zip | |
Fixed euler_a generator argument
Diffstat (limited to 'schedulers')
| -rw-r--r-- | schedulers/scheduling_euler_a.py | 3 |
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): | |||
| 198 | timestep: int, | 198 | timestep: int, |
| 199 | timestep_prev: int, | 199 | timestep_prev: int, |
| 200 | sample: torch.FloatTensor, | 200 | sample: torch.FloatTensor, |
| 201 | generator: None, | 201 | generator: torch.Generator = None, |
| 202 | return_dict: bool = True, | 202 | return_dict: bool = True, |
| 203 | ) -> Union[SchedulerOutput, Tuple]: | 203 | ) -> Union[SchedulerOutput, Tuple]: |
| 204 | """ | 204 | """ |
| @@ -240,7 +240,6 @@ class EulerAScheduler(SchedulerMixin, ConfigMixin): | |||
| 240 | sample_hat: torch.FloatTensor, | 240 | sample_hat: torch.FloatTensor, |
| 241 | sample_prev: torch.FloatTensor, | 241 | sample_prev: torch.FloatTensor, |
| 242 | derivative: torch.FloatTensor, | 242 | derivative: torch.FloatTensor, |
| 243 | generator: None, | ||
| 244 | return_dict: bool = True, | 243 | return_dict: bool = True, |
| 245 | ) -> Union[SchedulerOutput, Tuple]: | 244 | ) -> Union[SchedulerOutput, Tuple]: |
| 246 | """ | 245 | """ |