Pirog17000
7821135b0f
- Refactors KSamplers into `node_modules` for better organization. - Implements a robust tiling system in KSamplerMultiSeedPlus that uses gradient-mask inpainting to eliminate seams. - Fixes critical bugs related to VAE dimension mismatches, incorrect tensor shapes, and uneven image splits. - Corrects state pollution bugs, allowing for stable batch processing and chaining of multiple KSampler nodes. - Updates README with details on new tiling functionality and refactoring.
54 lines
2.2 KiB
Python
54 lines
2.2 KiB
Python
# Import enhanced KSampler classes from the node_modules package
|
|
from .node_modules.samplermultiseed import KSamplerMultiSeed, KSamplerMultiSeedPlus
|
|
|
|
# Import other classes from the main nodes.py file
|
|
from .nodes import StringCombine, Watermark, ImageScalePro, PromptRandomizer, DSLRNoise, TestResetButton, LensSimulatedBloom, CropImage, BatchCropFromMaskSimple, BatchUncropSimple, CropMaskByBBox, BlurMask, InvertMask, GradientMaskGenerator, ImageBlendByMask, BlurByMask, PreviewImageQueue
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"KSamplerMultiSeed": KSamplerMultiSeed,
|
|
"KSamplerMultiSeedPlus": KSamplerMultiSeedPlus,
|
|
"StringCombine": StringCombine,
|
|
"Watermark": Watermark,
|
|
"ImageScalePro": ImageScalePro,
|
|
"PromptRandomizer": PromptRandomizer,
|
|
"DSLRNoise": DSLRNoise,
|
|
"TestResetButton": TestResetButton,
|
|
"LensSimulatedBloom": LensSimulatedBloom,
|
|
"CropImage": CropImage,
|
|
"BatchCropFromMaskSimple": BatchCropFromMaskSimple,
|
|
"BatchUncropSimple": BatchUncropSimple,
|
|
"CropMaskByBBox": CropMaskByBBox,
|
|
"BlurMask": BlurMask,
|
|
"InvertMask": InvertMask,
|
|
"GradientMaskGenerator": GradientMaskGenerator,
|
|
"ImageBlendByMask": ImageBlendByMask,
|
|
"BlurByMask": BlurByMask,
|
|
"PreviewImageQueue": PreviewImageQueue
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"KSamplerMultiSeed": "KSampler (Multi-Seed)",
|
|
"KSamplerMultiSeedPlus": "KSampler (Multi-Seed+)",
|
|
"StringCombine": "Combine strings",
|
|
"Watermark": "Watermark",
|
|
"ImageScalePro": "Proportional Image Scaling",
|
|
"PromptRandomizer": "Prompt Randomizer",
|
|
"DSLRNoise": "DSLR Camera Noise",
|
|
"TestResetButton": "Test Reset Button",
|
|
"LensSimulatedBloom": "Lens-simulated Bloom",
|
|
"CropImage": "Crop Image Sides",
|
|
"BatchCropFromMaskSimple": "Crop From Mask",
|
|
"BatchUncropSimple": "Uncrop Image",
|
|
"CropMaskByBBox": "Crop Mask by BBox",
|
|
"BlurMask": "Blur Mask",
|
|
"InvertMask": "Invert Mask",
|
|
"GradientMaskGenerator": "Gradient Mask Generator",
|
|
"ImageBlendByMask": "Image Blend by Mask",
|
|
"BlurByMask": "Blur by Mask",
|
|
"PreviewImageQueue": "Preview Image (Queue)"
|
|
}
|
|
|
|
WEB_DIRECTORY = "./js"
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|