rename node

This commit is contained in:
spacepxl
2025-01-26 00:09:51 -05:00
parent 9584ef1de8
commit 58fe9ba700
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -1,7 +1,7 @@
from .nodes import *
NODE_CLASS_MAPPINGS = {
"Measure Loss": MeasureLoss,
"Measure Timestep Loss": MeasureTimestepLoss,
}
__all__ = ['NODE_CLASS_MAPPINGS']
+1 -1
View File
@@ -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 {