from local

This commit is contained in:
dohe0342 2023-04-11 16:36:55 +09:00
parent 0b87bff9a7
commit deb48a94a3
2 changed files with 2 additions and 1 deletions

View File

@ -79,7 +79,8 @@ class Transducer(nn.Module):
self.prompt = None
if prompt:
self.prompt = torch.randn((200, 512), requires_grad=True)
#self.prompt = torch.randn((200, 512), requires_grad=True)
self.prompt = torch.nn.Parameter(torch.randn((200, 512)))
def forward(
self,