Upload matryoshka.py
Browse files- scheduler/matryoshka.py +1 -1
scheduler/matryoshka.py
CHANGED
|
@@ -533,7 +533,7 @@ class MatryoshkaDDIMScheduler(SchedulerMixin, ConfigMixin):
|
|
| 533 |
|
| 534 |
def get_schedule_shifted(self, alpha_prod, scale_factor=None):
|
| 535 |
if (scale_factor is not None) and (scale_factor > 1): # rescale noise schedule
|
| 536 |
-
|
| 537 |
snr = alpha_prod / (1 - alpha_prod)
|
| 538 |
scaled_snr = snr / scale_factor
|
| 539 |
alpha_prod = 1 / (1 + 1 / scaled_snr)
|
|
|
|
| 533 |
|
| 534 |
def get_schedule_shifted(self, alpha_prod, scale_factor=None):
|
| 535 |
if (scale_factor is not None) and (scale_factor > 1): # rescale noise schedule
|
| 536 |
+
scale_factor = scale_factor ** self.schedule_shifted_power
|
| 537 |
snr = alpha_prod / (1 - alpha_prod)
|
| 538 |
scaled_snr = snr / scale_factor
|
| 539 |
alpha_prod = 1 / (1 + 1 / scaled_snr)
|