Wan2.1/ltx_video/utils/skip_layer_strategy.py

9 lines
169 B
Python

from enum import Enum, auto
class SkipLayerStrategy(Enum):
AttentionSkip = auto()
AttentionValues = auto()
Residual = auto()
TransformerBlock = auto()