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