Update src/diffusers/models/transformers/transformer_helios.py

Co-authored-by: YiYi Xu <yixu310@gmail.com>
This commit is contained in:
Shenghai Yuan
2026-02-26 09:18:36 +08:00
committed by GitHub
parent 31fb0cfdc5
commit f276d2ff84

View File

@@ -142,14 +142,6 @@ class HeliosAttnProcessor:
return hidden_states
class HeliosAttnProcessor2_0:
def __new__(cls, *args, **kwargs):
deprecation_message = (
"The HeliosAttnProcessor2_0 class is deprecated and will be removed in a future version. "
"Please use HeliosAttnProcessor instead. "
)
deprecate("HeliosAttnProcessor2_0", "1.0.0", deprecation_message, standard_warn=False)
return HeliosAttnProcessor(*args, **kwargs)
class HeliosAttention(torch.nn.Module, AttentionModuleMixin):