feat: add support for playground v2.5 (#3073)

* feat: add support for playground v2.5

* feat: add preset for playground v2.5

* feat: change URL to mashb1t

* feat: optimize playground v2.5 preset
This commit is contained in:
Manuel Schmid
2024-06-04 20:15:49 +02:00
committed by GitHub
parent cb24c686b0
commit 2d55a5f257
11 changed files with 233 additions and 30 deletions
-2
View File
@@ -51,8 +51,6 @@ def patched_register_schedule(self, given_betas=None, beta_schedule="linear", ti
self.linear_end = linear_end
sigmas = torch.tensor(((1 - alphas_cumprod) / alphas_cumprod) ** 0.5, dtype=torch.float32)
self.set_sigmas(sigmas)
alphas_cumprod = torch.tensor(alphas_cumprod, dtype=torch.float32)
self.set_alphas_cumprod(alphas_cumprod)
return