File size: 1,912 Bytes
3a6b18a | 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 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 | from .krea2_reference import (
DEFAULT_GROUNDING_PX,
REFERENCE_MASK_MODE_DEEMPHASIZE,
REFERENCE_MASK_MODE_EXCLUDE,
REFERENCE_MASK_MODE_EXCLUDE_BLANK,
REFERENCE_MASK_MODES,
REFERENCE_MODE_APPEND,
REFERENCE_MODE_OFF,
REFERENCE_MODE_PREPEND,
REFERENCE_MODES,
Krea2AppendReferenceImagesStep,
Krea2AppendReferenceTextEncoderStep,
Krea2AppendReferenceWorkflow,
Krea2PrependReferenceImagesStep,
Krea2PrependReferenceTextEncoderStep,
Krea2PrependReferenceWorkflow,
Krea2ReferenceAutoBlocks,
Krea2ReferenceLatentsStep,
Krea2ReferenceTextInputsStep,
Krea2TurboAppendReferenceWorkflow,
Krea2TurboPrependReferenceWorkflow,
Krea2TurboReferenceAutoBlocks,
Krea2TurboVisionReferenceWorkflow,
Krea2VisionReferenceImagesStep,
Krea2VisionReferenceTextEncoderStep,
Krea2VisionReferenceWorkflow,
)
__all__ = [
# Top level: `reference_mode` selects the workflow.
"Krea2ReferenceAutoBlocks",
"Krea2TurboReferenceAutoBlocks",
# Per-mode workflows.
"Krea2VisionReferenceWorkflow",
"Krea2AppendReferenceWorkflow",
"Krea2PrependReferenceWorkflow",
"Krea2TurboVisionReferenceWorkflow",
"Krea2TurboAppendReferenceWorkflow",
"Krea2TurboPrependReferenceWorkflow",
# Leaf blocks.
"Krea2VisionReferenceImagesStep",
"Krea2AppendReferenceImagesStep",
"Krea2PrependReferenceImagesStep",
"Krea2VisionReferenceTextEncoderStep",
"Krea2AppendReferenceTextEncoderStep",
"Krea2PrependReferenceTextEncoderStep",
"Krea2ReferenceTextInputsStep",
"Krea2ReferenceLatentsStep",
# Mode constants.
"DEFAULT_GROUNDING_PX",
"REFERENCE_MODES",
"REFERENCE_MODE_APPEND",
"REFERENCE_MODE_OFF",
"REFERENCE_MODE_PREPEND",
"REFERENCE_MASK_MODES",
"REFERENCE_MASK_MODE_DEEMPHASIZE",
"REFERENCE_MASK_MODE_EXCLUDE",
"REFERENCE_MASK_MODE_EXCLUDE_BLANK",
]
|