mirror of
https://github.com/huggingface/diffusers.git
synced 2026-03-06 00:31:48 +08:00
Update src/diffusers/models/transformers/transformer_helios.py
Co-authored-by: YiYi Xu <yixu310@gmail.com>
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user