- Viewer: send step images as temp files instead of base64 (dead send_json call removed; no history/WebSocket bloat) - Sampler: early stop returns the latest x0 and always saves the stop step; PSNR inf counts as converged - Sampler: ignore auto_stop when stop_threshold is unsuitable for PSNR/SSIM (avoids stopping at step 2) - Sampler: keep step images as uint8 in a local list; pop downscale_ratio_spacial; compat with older fix_empty_latent_channels - Fix display name mapping for the sampler node - Viewer JS: larger slider handle/track, wheel and keyboard stepping, clear timer on node removal, validate play speed - README: fix typos and document threshold semantics / controls Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
17 lines
560 B
Python
17 lines
560 B
Python
from .StepByStep_Sampler import StepByStepSampler
|
|
from .StepByStep_Viewer import StepStepPlayer, StepStepComparer
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"StepByStepSampler": StepByStepSampler,
|
|
"StepStepPlayer": StepStepPlayer,
|
|
"StepStepComparer": StepStepComparer
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"StepByStepSampler": "Step-by-Step Sampler",
|
|
"StepStepPlayer": "Step-by-Step Player",
|
|
"StepStepComparer": "Step-by-Step Comparer"
|
|
}
|
|
|
|
WEB_DIRECTORY = "./js"
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"] |