Spaces:
Running on Zero
Running on Zero
File size: 1,361 Bytes
b701455 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 | """Processors module for LightDiffusion-Next.
This module provides post-processing components that can be applied
to generated images in a clean, modular fashion:
- HiresFix: High-resolution upscaling with latent re-diffusion
- Adetailer: Automatic face/body enhancement
- Img2Img: Image-to-image generation and upscaling
- AutoHDRProcessor: HDR effect enhancement
- StableFastProcessor: Model compilation optimization
- DeepCacheProcessor: U-Net feature caching
All processors follow the same interface:
- is_enabled(ctx) -> bool: Check if processor should run
- process(ctx, model, **kwargs) -> ctx: Apply processing
Usage:
from src.Processors import HiresFix, Adetailer, AutoHDRProcessor
# Check and apply
if HiresFix.is_enabled(ctx):
latents = HiresFix.apply(latents, ctx, model, positive, negative)
# Or use the unified process method
ctx = AutoHDRProcessor.process(ctx, model)
"""
from src.Processors.HiresFix import HiresFix
from src.Processors.Adetailer import Adetailer
from src.Processors.Img2Img import Img2Img
from src.Processors.AutoHDRProcessor import AutoHDRProcessor
from src.Processors.StableFastProcessor import StableFastProcessor, DeepCacheProcessor
__all__ = [
"HiresFix",
"Adetailer",
"Img2Img",
"AutoHDRProcessor",
"StableFastProcessor",
"DeepCacheProcessor",
]
|