File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
src/diffusers/models/autoencoders Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -67,7 +67,7 @@ def __init__(
6767 norm = "bn2d" ,
6868 act_func = "relu" ,
6969 ):
70- super (ConvLayer , self ).__init__ ()
70+ super ().__init__ ()
7171
7272 padding = kernel_size // 2
7373 padding *= dilation
@@ -357,7 +357,7 @@ def __init__(
357357 context_module : str = "LiteMLA" ,
358358 local_module : str = "MBConv" ,
359359 ):
360- super (EfficientViTBlock , self ).__init__ ()
360+ super ().__init__ ()
361361 if context_module == "LiteMLA" :
362362 self .context_module = ResidualBlock (
363363 LiteMLA (
@@ -406,7 +406,7 @@ def __init__(
406406 post_act = None ,
407407 pre_norm : Optional [nn .Module ] = None ,
408408 ):
409- super (ResidualBlock , self ).__init__ ()
409+ super ().__init__ ()
410410
411411 self .pre_norm = pre_norm
412412 self .main = main
@@ -433,7 +433,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
433433
434434class OpSequential (nn .Module ):
435435 def __init__ (self , op_list : list [Optional [nn .Module ]]):
436- super (OpSequential , self ).__init__ ()
436+ super ().__init__ ()
437437 valid_op_list = []
438438 for op in op_list :
439439 if op is not None :
You can’t perform that action at this time.
0 commit comments