File size: 753 Bytes
497c818 | 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 | """Minimal extracted implementation for the user's PixelDiT three-control method."""
from .independent_gated_control import (
CONTROL_NAMES,
IndependentBranchGatedFusion,
StructureAwareGatedZeroAdapter,
apply_multi_control_mode,
mask_inactive_control_grads,
mode_to_keep,
sample_control_mode_ddp,
sobel_structure_map,
)
from .losses import MultiConditionCycleLoss, SoftCannyImagePyramidCycleLoss
__all__ = [
"CONTROL_NAMES",
"IndependentBranchGatedFusion",
"StructureAwareGatedZeroAdapter",
"apply_multi_control_mode",
"mask_inactive_control_grads",
"mode_to_keep",
"sample_control_mode_ddp",
"sobel_structure_map",
"MultiConditionCycleLoss",
"SoftCannyImagePyramidCycleLoss",
]
|