Skip to content

Commit 2c470fc

Browse files
committed
Fix SeedVR2 3B RoPE buffer loading
1 parent 07552db commit 2c470fc

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

comfy/ldm/seedvr/model.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1092,7 +1092,6 @@ def __init__(
10921092
norm_eps: float,
10931093
ada,
10941094
qk_bias: bool,
1095-
qk_rope: Optional[bool],
10961095
qk_norm,
10971096
mlp_type: str,
10981097
shared_weights: bool,
@@ -1115,7 +1114,7 @@ def __init__(
11151114
qk_bias=qk_bias,
11161115
qk_norm=qk_norm,
11171116
qk_norm_eps=norm_eps,
1118-
rope_type=rope_type if qk_rope else None,
1117+
rope_type=rope_type,
11191118
rope_dim=rope_dim,
11201119
shared_weights=shared_weights,
11211120
window=kwargs.pop("window", None),

0 commit comments

Comments
 (0)