Skip to content

GitLab

"docs/source/en/model_doc/convnext.mdx" did not exist on "c47d2592413e24a38e65917a6d5f265fc748e8a7"
Unverified Commit 53194991 authored by NielsRogge's avatar NielsRogge Committed by GitHub
Browse files

[Mask2Former] Remove SwinConfig (#24259) 

Remove SwinConfig
parent fb6a6276
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 5 deletions
src/transformers/models/mask2former/modeling_mask2former.py
View file @ 53194991
......@@ -23,7 +23,7 @@ import numpy as np
import torch
from torch import Tensor, nn
from ... import AutoBackbone, SwinConfig
from ... import AutoBackbone
from ...activations import ACT2FN
from ...file_utils import (
ModelOutput,
......@@ -1388,10 +1388,7 @@ class Mask2FormerPixelLevelModule(nn.Module):
"""
super().__init__()
backbone_config_dict = config.backbone_config.to_dict()
backbone_config = SwinConfig.from_dict(backbone_config_dict)
self.encoder = AutoBackbone.from_config(backbone_config)
self.encoder = AutoBackbone.from_config(config.backbone_config)
self.decoder = Mask2FormerPixelDecoder(config, feature_channels=self.encoder.channels)
def forward(self, pixel_values: Tensor, output_hidden_states: bool = False) -> Mask2FormerPixelLevelModuleOutput:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment