from local

This commit is contained in:
dohe0342 2023-04-26 12:02:04 +09:00
parent 90e1d2299e
commit 1e0d896e68
2 changed files with 1 additions and 0 deletions

View File

@ -492,6 +492,7 @@ class Data2VecAudioModel(BaseFairseqModel):
if prompt is not None:
print(features.size())
print(padding_mask.size())
print(padding_mask.nonzero())
exit()
prompt = prompt.expand((features.size()[0], prompt.size()[0], prompt.size()[1]))
features = torch.cat([prompt, features], dim=1)