Update ml_decoder.py

huggingface · Dec 26, 2024 · b927237 · b927237
1 parent abbdf05
commit b927237
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/timm/layers/ml_decoder.py b/timm/layers/ml_decoder.py
@@ -349,7 +349,7 @@ def __init__(
         learnable_class_embed: bool = False,
         embed_drop: float = 0.1,
         embed_norm: bool = True,
-        k_norm: bool = False,
+        qk_norm: bool = False,
         attn_drop: float = 0.1,
         mlp_ratio: float = 8/3,
         proj_drop: float = 0.1,