diff --git a/timm/models/hieradet_sam2.py b/timm/models/hieradet_sam2.py index fbd7ce28ec..cd8bc6c8e9 100644 --- a/timm/models/hieradet_sam2.py +++ b/timm/models/hieradet_sam2.py @@ -353,6 +353,7 @@ def __init__( window_size=window_size, norm_layer=norm_layer, act_layer=act_layer, + init_values=init_values, ) embed_dim = dim_out