from local

This commit is contained in:
dohe0342 2023-04-11 15:54:00 +09:00
parent 6476c0715b
commit 4d5560e194
2 changed files with 1 additions and 0 deletions

View File

@ -498,6 +498,7 @@ class Data2VecAudioModel(BaseFairseqModel):
print(features.size())
prompt_padding_mask = torch.zeros(prompt.size()[0], prompt.size()[1]).type(torch.BoolTensor).to(features.device)
print(prompt_padding_mask.size())
print(padding_mask.size())
padding_mask = torch.cat([prompt_padding_mask, padding_mask])
print(padding_mask.size())