| """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", |
| ] |
|
|
