WANGP1 / ltx_video /utils /skip_layer_strategy.py
rahul7star's picture
Migrated from GitHub
30f8a30 verified
raw
history blame contribute delete
169 Bytes
from enum import Enum, auto
class SkipLayerStrategy(Enum):
AttentionSkip = auto()
AttentionValues = auto()
Residual = auto()
TransformerBlock = auto()