From 63636c3355758fbe819d3c55231201c59cfc0bf2 Mon Sep 17 00:00:00 2001
From: Max Tretikov <max@tretikov.com>
Date: Thu, 13 Jun 2024 15:15:30 -0600
Subject: [PATCH] Fix no-member of EPS inherited classes

---
 comfy/model_sampling.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/comfy/model_sampling.py b/comfy/model_sampling.py
index 6c0f7b7da..7bdd3636d 100644
--- a/comfy/model_sampling.py
+++ b/comfy/model_sampling.py
@@ -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