|  | 
| 21 | 21 | 
 | 
| 22 | 22 | from ...configuration_utils import ConfigMixin, register_to_config | 
| 23 | 23 | from ...loaders import FromOriginalModelMixin, PeftAdapterMixin | 
| 24 |  | -from ...utils import USE_PEFT_BACKEND, logging, scale_lora_layers, unscale_lora_layers | 
|  | 24 | +from ...utils import USE_PEFT_BACKEND, deprecate, logging, scale_lora_layers, unscale_lora_layers | 
| 25 | 25 | from ...utils.torch_utils import maybe_allow_in_graph | 
| 26 | 26 | from ..attention import AttentionMixin, AttentionModuleMixin, FeedForward | 
| 27 | 27 | from ..attention_dispatch import dispatch_attention_fn | 
| @@ -164,6 +164,16 @@ def apply_rotary_emb( | 
| 164 | 164 |         return hidden_states | 
| 165 | 165 | 
 | 
| 166 | 166 | 
 | 
|  | 167 | +class SkyReelsV2AttnProcessor2_0: | 
|  | 168 | +    def __new__(cls, *args, **kwargs): | 
|  | 169 | +        deprecation_message = ( | 
|  | 170 | +            "The SkyReelsV2AttnProcessor2_0 class is deprecated and will be removed in a future version. " | 
|  | 171 | +            "Please use SkyReelsV2AttnProcessor instead. " | 
|  | 172 | +        ) | 
|  | 173 | +        deprecate("SkyReelsV2AttnProcessor2_0", "1.0.0", deprecation_message, standard_warn=False) | 
|  | 174 | +        return SkyReelsV2AttnProcessor(*args, **kwargs) | 
|  | 175 | + | 
|  | 176 | + | 
| 167 | 177 | class SkyReelsV2Attention(torch.nn.Module, AttentionModuleMixin): | 
| 168 | 178 |     _default_processor_cls = SkyReelsV2AttnProcessor | 
| 169 | 179 |     _available_processors = [SkyReelsV2AttnProcessor] | 
|  | 
0 commit comments