diff --git a/timm/models/dependencyvit.py b/timm/models/dependencyvit.py index 4f297fde5..059b78e1d 100644 --- a/timm/models/dependencyvit.py +++ b/timm/models/dependencyvit.py @@ -244,7 +244,7 @@ def __init__( qkv_bias=False, init_values=1e-6, fc_norm=False, - pos_embed='none', + pos_embed=pos_embed, ) if prune_layers is not None: