Skip to content

Commit 93d859c

Browse files
authored
Fix type annotation syntax in MotionEncoder_tc __init__ (#10186)
## Summary Fixed incorrect type hint syntax in `MotionEncoder_tc.__init__()` parameter list. ## Changes - Line 647: Changed `num_heads=int` to `num_heads: int` - This corrects the parameter annotation from a default value assignment to proper type hint syntax ## Details The parameter was using assignment syntax (`=`) instead of type annotation syntax (`:`), which would incorrectly set the default value to the `int` class itself rather than annotating the expected type.
1 parent 4614ee0 commit 93d859c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

comfy/ldm/wan/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -903,7 +903,7 @@ class MotionEncoder_tc(nn.Module):
903903
def __init__(self,
904904
in_dim: int,
905905
hidden_dim: int,
906-
num_heads=int,
906+
num_heads: int,
907907
need_global=True,
908908
dtype=None,
909909
device=None,

0 commit comments

Comments
 (0)