Merge branch 'master' into flipflop-stream

This commit is contained in:
Jedrzej Kosinski
2025-10-03 14:32:56 -07:00
15 changed files with 559 additions and 446 deletions

View File

@@ -909,7 +909,7 @@ class MotionEncoder_tc(nn.Module):
def __init__(self,
in_dim: int,
hidden_dim: int,
num_heads=int,
num_heads: int,
need_global=True,
dtype=None,
device=None,