from local

This commit is contained in:
dohe0342 2023-04-11 16:25:31 +09:00
parent dcfd8fb6a8
commit 8c7c3a0171
3 changed files with 3 additions and 2 deletions

View File

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