Skip to content

Commit cbec58d

Browse files
authored
fix broken model constructor (#11)
1 parent ac3930b commit cbec58d

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

src/models/ac_predictor.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -195,12 +195,6 @@ def vit_ac_predictor(**kwargs):
195195
mlp_ratio=4,
196196
qkv_bias=True,
197197
norm_layer=partial(nn.LayerNorm, eps=1e-6),
198-
depth=24,
199-
predictor_embed_dim=1024,
200-
num_heads=16,
201-
is_frame_causal=True,
202-
use_extrinsics=False,
203-
uniform_power=True,
204198
**kwargs
205199
)
206200
return model

0 commit comments

Comments
 (0)