Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

lint

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-09-15 18:01:34 +02:00
parent fe6c3acd16
commit a941d101c5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/minimax/minimax_vdn.py
+1 -1
View File
@@ -52,7 +52,7 @@ class VectorizedGatedDeltaBranch(nn.Module):
error = v_f - pred_v
# In-place state update avoids intermediate 5D tensor allocations
state.mul_(1.0 - b_f).addcmul_(b_f, error.unsqueeze(-1), k_f.unsqueeze(-2))
state.mul_(1.0 - b_f).addcmul_(b_f, error.unsqueeze(-1), k_f.unsqueeze(-2)) # ty: ignore[too-many-positional-arguments]
frame_out = torch.matmul(q_f.unsqueeze(-2), state).squeeze(-2)