Update dependencyvit.py

huggingface · Dec 31, 2024 · f63d2e8 · f63d2e8
1 parent 1aa5ddf
commit f63d2e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: