"""
FLUX.2 Klein Conditioning Enhancer - Fixed Version

Architecture:
- Shape: [batch, 512, 12288]
- Active region: Dynamic (from attention_mask, typically ~77 tokens)
- Padding: Remaining positions (low variance ~2.3)
- Metadata: pooled_output, attention_mask, reference_latents (edit mode)

FIXED: Previous version had mean-recentering that undid the scaling.
This version uses direct operations with measurable effects.
"""

from .flux2_klein_enhancer import (
    Flux2KleinEnhancer,
    Flux2KleinDetailController,
)

NODE_CLASS_MAPPINGS = {
    "Flux2KleinEnhancer": Flux2KleinEnhancer,
    "Flux2KleinDetailController": Flux2KleinDetailController,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "Flux2KleinEnhancer": "FLUX.2 Klein Enhancer",
    "Flux2KleinDetailController": "FLUX.2 Klein Detail Controller",
}

__version__ = "2.1.0"