remove unused is_decoder parameter in DetrAttention (#24226)

* issue#24161 remove unused is_decoder parameter in DetrAttention

* #24161 fix check_repository_consistency fail
This commit is contained in:
JayL0321 2023-06-15 03:39:32 -07:00 committed by GitHub
parent 33196b459c
commit a611ac9b3f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 0 additions and 10 deletions

View File

@ -534,7 +534,6 @@ class DetrAttention(nn.Module):
embed_dim: int,
num_heads: int,
dropout: float = 0.0,
is_decoder: bool = False,
bias: bool = True,
):
super().__init__()

View File

@ -499,7 +499,6 @@ class DetrAttention(nn.Module):
embed_dim: int,
num_heads: int,
dropout: float = 0.0,
is_decoder: bool = False,
bias: bool = True,
):
super().__init__()
@ -697,7 +696,6 @@ class DetrDecoderLayer(nn.Module):
embed_dim=self.embed_dim,
num_heads=config.decoder_attention_heads,
dropout=config.attention_dropout,
is_decoder=True,
)
self.dropout = config.dropout
self.activation_fn = ACT2FN[config.activation_function]
@ -708,7 +706,6 @@ class DetrDecoderLayer(nn.Module):
self.embed_dim,
config.decoder_attention_heads,
dropout=config.attention_dropout,
is_decoder=True,
)
self.encoder_attn_layer_norm = nn.LayerNorm(self.embed_dim)
self.fc1 = nn.Linear(self.embed_dim, config.decoder_ffn_dim)

View File

@ -416,7 +416,6 @@ class DetrAttention(nn.Module):
embed_dim: int,
num_heads: int,
dropout: float = 0.0,
is_decoder: bool = False,
bias: bool = True,
):
super().__init__()
@ -545,7 +544,6 @@ class DetrDecoderLayer(nn.Module):
embed_dim=self.embed_dim,
num_heads=config.decoder_attention_heads,
dropout=config.attention_dropout,
is_decoder=True,
)
self.dropout = config.dropout
self.activation_fn = ACT2FN[config.activation_function]
@ -556,7 +554,6 @@ class DetrDecoderLayer(nn.Module):
self.embed_dim,
config.decoder_attention_heads,
dropout=config.attention_dropout,
is_decoder=True,
)
self.encoder_attn_layer_norm = nn.LayerNorm(self.embed_dim)
self.fc1 = nn.Linear(self.embed_dim, config.decoder_ffn_dim)

View File

@ -440,7 +440,6 @@ class TableTransformerAttention(nn.Module):
embed_dim: int,
num_heads: int,
dropout: float = 0.0,
is_decoder: bool = False,
bias: bool = True,
):
super().__init__()
@ -642,7 +641,6 @@ class TableTransformerDecoderLayer(nn.Module):
embed_dim=self.embed_dim,
num_heads=config.decoder_attention_heads,
dropout=config.attention_dropout,
is_decoder=True,
)
self.dropout = config.dropout
self.activation_fn = ACT2FN[config.activation_function]
@ -653,7 +651,6 @@ class TableTransformerDecoderLayer(nn.Module):
self.embed_dim,
config.decoder_attention_heads,
dropout=config.attention_dropout,
is_decoder=True,
)
self.encoder_attn_layer_norm = nn.LayerNorm(self.embed_dim)
self.fc1 = nn.Linear(self.embed_dim, config.decoder_ffn_dim)