from local

This commit is contained in:
dohe0342 2023-01-03 16:55:31 +09:00
parent 841d62f98c
commit 0e9358f601
2 changed files with 1 additions and 1 deletions

View File

@ -160,7 +160,7 @@ class ResidualAdapterModule(nn.Module):
)
self.adapter_layers = nn.ModuleList(
[build_adapter(embedding_dim, proj_dim, type_='conv') for _ in range(layer_num)]
[build_adapter(embedding_dim, proj_dim, type_=self.type) for _ in range(layer_num)]
)
def forward(self, x, layer_id=-1):