Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fooocus GitHub Bot Commit

Browse Source 
This commit is generated by a GitHub bot of Fooocus
This commit is contained in:
lllyasviel
2023-10-20 02:43:05 -07:00
parent 04066124c2
commit f1e1d2f7d7
4 changed files with 34 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/headless/fcbh/k_diffusion/external.py
+4
View File
@@ -97,6 +97,10 @@ class DiscreteSchedule(nn.Module):
input = input * ((utils.append_dims(sigma, input.ndim) ** 2 + 1.0) ** 0.5)
return (input - self(input, sigma, **kwargs)) / utils.append_dims(sigma, input.ndim)
def predict_eps_sigma(self, input, sigma, **kwargs):
input = input * ((utils.append_dims(sigma, input.ndim) ** 2 + 1.0) ** 0.5)
return (input - self(input, sigma, **kwargs)) / utils.append_dims(sigma, input.ndim)
class DiscreteEpsDDPMDenoiser(DiscreteSchedule):
"""A wrapper for discrete schedule DDPM models that output eps (the predicted
noise)."""