Skip to content

Commit 1c55452

Browse files
committed
removed diffusers import
1 parent e18128c commit 1c55452

1 file changed

Lines changed: 3 additions & 5 deletions

File tree

src/maxdiffusion/models/ltx_video/transformers/activations.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,6 @@
55
from flax import linen as nn
66
from flax.linen.initializers import lecun_normal
77

8-
from diffusers.utils.deprecation_utils import deprecate
9-
108
from maxdiffusion.models.ltx_video.linear import DenseGeneral, KernelInitializer
119

1210

@@ -117,9 +115,9 @@ class GEGLU(nn.Module):
117115

118116
@nn.compact
119117
def __call__(self, hidden_states, *args, **kwargs):
120-
if len(args) > 0 or kwargs.get("scale", None) is not None:
121-
deprecation_message = "The `scale` argument is deprecated and will be ignored. Please remove it, as passing it will raise an error in the future. `scale` should directly be passed while calling the underlying pipeline component i.e., via `cross_attention_kwargs`."
122-
deprecate("scale", "1.0.0", deprecation_message)
118+
# if len(args) > 0 or kwargs.get("scale", None) is not None:
119+
# deprecation_message = "The `scale` argument is deprecated and will be ignored. Please remove it, as passing it will raise an error in the future. `scale` should directly be passed while calling the underlying pipeline component i.e., via `cross_attention_kwargs`."
120+
# deprecate("scale", "1.0.0", deprecation_message)
123121

124122
proj = DenseGeneral(
125123
features=self.dim_out * 2,

0 commit comments

Comments
 (0)