From 58fe9ba7004eda8d4ea1981bd33be9df5d1bc9de Mon Sep 17 00:00:00 2001
From: spacepxl <spacetug93@gmail.com>
Date: Sun, 26 Jan 2025 00:09:51 -0500
Subject: [PATCH] rename node

---
 __init__.py | 2 +-
 nodes.py    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/__init__.py b/__init__.py
index a740c7a..3d11af2 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,7 +1,7 @@
 from .nodes import *
 
 NODE_CLASS_MAPPINGS = {
-    "Measure Loss": MeasureLoss,
+    "Measure Timestep Loss": MeasureTimestepLoss,
 }
 
 __all__ = ['NODE_CLASS_MAPPINGS']
\ No newline at end of file
diff --git a/nodes.py b/nodes.py
index 524e20b..86ce5eb 100644
--- a/nodes.py
+++ b/nodes.py
@@ -37,7 +37,7 @@ def common_ksampler(model, seed, steps, scheduler, positive, latent, denoise=1.0
     loss = F.mse_loss(samples, latent_image, reduction="none").mean(dim=(1,2,3))
     return loss
 
-class MeasureLoss:
+class MeasureTimestepLoss:
     @classmethod
     def INPUT_TYPES(s):
         return {