from local

This commit is contained in:
dohe0342 2022-12-26 14:18:39 +09:00
parent 6f6e91f992
commit b8232df095
2 changed files with 1 additions and 1 deletions

View File

@ -153,7 +153,7 @@ class ResidualAdapterModule(nn.Module):
[build_adapter(embedding_dim, proj_dim) for _ in range(layer_num)]
)
def forward(x, layer_id):
def forward(x, layer_id=-1):
x = x.transpose(0, 1)
residual = x
x = self.adapter_layers[layer_id](x)