conditioning/conditioning_method.py

from enum import Enum


class ConditioningMethod(Enum):

    INPUT_PREPEND = "input_prepend"
    INPUT_SUM = "input_sum"
    CROSS_ATTENTION = "cross_attention"


if __name__ == "__main__":
    c = ConditioningMethod.INPUT_PREPEND
    print(f'{c=}')
    print(f'{c.value=}')