hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 03:10:21 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #14145 from drhead/zero-terminal-snr

Browse Source 
Implement zero terminal SNR noise schedule option
This commit is contained in:
AUTOMATIC1111
2024-01-01 14:45:12 +03:00
committed by GitHub
parent d613cd17c7 5381405eaa
commit 267fd5d76b
4 changed files with 37 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_samplers_timesteps.py
View File

@@ -36,7 +36,7 @@ class CompVisTimestepsVDenoiser(torch.nn.Module):
self.inner_model = model
def predict_eps_from_z_and_v(self, x_t, t, v):
return self.inner_model.sqrt_alphas_cumprod[t.to(torch.int), None, None, None] * v + self.inner_model.sqrt_one_minus_alphas_cumprod[t.to(torch.int), None, None, None] * x_t
return torch.sqrt(self.inner_model.alphas_cumprod)[t.to(torch.int), None, None, None] * v + torch.sqrt(1 - self.inner_model.alphas_cumprod)[t.to(torch.int), None, None, None] * x_t
def forward(self, input, timesteps, **kwargs):
model_output = self.inner_model.apply_model(input, timesteps, **kwargs)