from local

This commit is contained in:
dohe0342 2023-04-11 16:26:09 +09:00
parent 8c7c3a0171
commit 27173eb413
2 changed files with 1 additions and 0 deletions

View File

@ -77,6 +77,7 @@ class Transducer(nn.Module):
nn.LogSoftmax(dim=-1),
)
self.prompt = None
if prompt:
self.prompt = torch.randn((50, 512), requires_grad=True)