from local

This commit is contained in:
dohe0342 2023-04-11 15:19:43 +09:00
parent 3b1ca50e6f
commit 895bd08ddc
2 changed files with 2 additions and 0 deletions

View File

@ -492,6 +492,8 @@ class Data2VecAudioModel(BaseFairseqModel):
if prompt is not None:
#features = torch.cat([features, prompt])
features = torch.cat([prompt, features])
prompt_padding_mask = torch.zeros(promt.size()).type(torch.BoolTensor).to(features.device)
padding_mask
features = self.layer_norm(features)