Update ml_decoder.py

huggingface · Dec 29, 2024 · 5c32658 · 5c32658
1 parent 44f93e9
commit 5c32658
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/timm/layers/ml_decoder.py b/timm/layers/ml_decoder.py
@@ -365,7 +365,7 @@ def __init__(
         self.query_embed = None
         self.query_dim = 0
         self.shared_fc = shared_fc
-        num_groups = num_classes if num_groups < 1
+        num_groups = num_classes if num_groups < 1 else num_groups
 
         # case using class embed
         if have_class_embed: