riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

[Mask2Former] Remove SwinConfig (#24259)

Browse Source 
Remove SwinConfig
This commit is contained in:
NielsRogge 2023-06-27 19:33:55 +02:00 committed by GitHub
parent fb6a62762f
commit 53194991e9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/transformers/models/mask2former/modeling_mask2former.py
View File

@ -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: