cleanup redundant edit

This commit is contained in:
Alex "mcmonkey" Goodwin 2024-09-16 12:40:05 +09:00 committed by GitHub
parent d11b29d036
commit 5e83837c01
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -22,7 +22,7 @@ class FeedForward(nn.Module):
super().__init__() super().__init__()
inner_dim = int(dim * mult) inner_dim = int(dim * mult)
if not dim_out: if not dim_out:
dim_out = dim() if isfunction(dim) else dim dim_out = dim
project_in = nn.Sequential( project_in = nn.Sequential(
ops.Linear(dim, inner_dim), ops.Linear(dim, inner_dim),
nn.GELU() nn.GELU()