wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-10 14:20:49 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Fix no-member of EPS inherited classes

Browse Source
This commit is contained in:
Max Tretikov 2024-06-13 15:15:30 -06:00
parent c69d4cae0a
commit 63636c3355
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_sampling.py
View File

@ -3,6 +3,8 @@ from .ldm.modules.diffusionmodules.util import make_beta_schedule
import math
class EPS:
sigma_data: float = None
def calculate_input(self, sigma, noise):
sigma = sigma.view(sigma.shape[:1] + (1,) * (noise.ndim - 1))
return noise / (sigma ** 2 + self.sigma_data ** 2) ** 0.5