wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-30 00:00:26 +08:00
Code Issues Actions 20 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 import math
class EPS: class EPS:
sigma_data: float = None
def calculate_input(self, sigma, noise): def calculate_input(self, sigma, noise):
sigma = sigma.view(sigma.shape[:1] + (1,) * (noise.ndim - 1)) sigma = sigma.view(sigma.shape[:1] + (1,) * (noise.ndim - 1))
return noise / (sigma ** 2 + self.sigma_data ** 2) ** 0.5 return noise / (sigma ** 2 + self.sigma_data ** 2) ** 0.5