src/models/flash_head/ltx_video/utils/skip_layer_strategy.py

169 bytes · 9 lines · capsule://quake0day/[email protected] raw on github

from enum import Enum, auto


class SkipLayerStrategy(Enum):
    AttentionSkip = auto()
    AttentionValues = auto()
    Residual = auto()
    TransformerBlock = auto()