Skip to content

Commit 336253f

Browse files
committed
remove unused argument in WindowSDPA
1 parent 31b5f80 commit 336253f

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/refiners/foundationals/swin/swin_transformer.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -173,7 +173,6 @@ def forward(self) -> Tensor:
173173
class WindowSDPA(fl.Module):
174174
def __init__(
175175
self,
176-
dim: int,
177176
window_size: int,
178177
num_heads: int,
179178
shift: bool = False,
@@ -223,7 +222,7 @@ def __init__(
223222
) -> None:
224223
super().__init__(
225224
fl.Linear(dim, dim * 3, bias=True, device=device),
226-
WindowSDPA(dim, window_size, num_heads, shift, device=device),
225+
WindowSDPA(window_size, num_heads, shift, device=device),
227226
fl.Linear(dim, dim, device=device),
228227
)
229228

0 commit comments

Comments
 (0)