Skip to content

Commit cde3a7d

Browse files
committed
ParallelScalingBlock needs extra args even if not used
1 parent bc1085b commit cde3a7d

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

timm/models/vision_transformer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -300,7 +300,9 @@ def __init__(
300300
drop_path: float = 0.,
301301
act_layer: Type[nn.Module] = nn.GELU,
302302
norm_layer: Type[nn.Module] = LayerNorm,
303-
mlp_layer: Optional[Type[nn.Module]] = None,
303+
mlp_layer: Optional[Type[nn.Module]] = None, # not used
304+
attn_layer: Optional[LayerType] = None, # not used
305+
depth: int = 0, # not used
304306
device = None,
305307
dtype = None,
306308
) -> None:

0 commit comments

Comments
 (0)