Fix: Rename keyword argument in_channels to num_channels (#35289)

Fix: Rename keyword argument in_channels to num_channels in some default backbone configs
huggingface · Dec 27, 2024 · 7f97d01 · 7f97d01
1 parent 4eb17b2
commit 7f97d01
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/transformers/models/mask2former/configuration_mask2former.py b/src/transformers/models/mask2former/configuration_mask2former.py
@@ -171,7 +171,7 @@ def __init__(
             logger.info("`backbone_config` is `None`. Initializing the config with the default `Swin` backbone.")
             backbone_config = CONFIG_MAPPING["swin"](
                 image_size=224,
-                in_channels=3,
+                num_channels=3,
                 patch_size=4,
                 embed_dim=96,
                 depths=[2, 2, 18, 2],

diff --git a/src/transformers/models/maskformer/configuration_maskformer.py b/src/transformers/models/maskformer/configuration_maskformer.py
@@ -131,7 +131,7 @@ def __init__(
             # fall back to https://huggingface.co/microsoft/swin-base-patch4-window12-384-in22k
             backbone_config = SwinConfig(
                 image_size=384,
-                in_channels=3,
+                num_channels=3,
                 patch_size=4,
                 embed_dim=128,
                 depths=[2, 2, 18, 2],

diff --git a/src/transformers/models/oneformer/configuration_oneformer.py b/src/transformers/models/oneformer/configuration_oneformer.py
@@ -201,7 +201,7 @@ def __init__(
             logger.info("`backbone_config` is unset. Initializing the config with the default `Swin` backbone.")
             backbone_config = CONFIG_MAPPING["swin"](
                 image_size=224,
-                in_channels=3,
+                num_channels=3,
                 patch_size=4,
                 embed_dim=96,
                 depths=[2, 2, 6, 2],