from local

This commit is contained in:
dohe0342 2023-04-21 10:22:25 +09:00
parent 2937aaec0b
commit ef41a98cf6
4 changed files with 1 additions and 1 deletions

Binary file not shown.

Binary file not shown.

View File

@ -80,7 +80,7 @@ class Transducer(nn.Module):
self.prompt = None self.prompt = None
if prompt: 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((30, 512))) self.prompt = torch.nn.Parameter(torch.randn((20, 512)))
def forward( def forward(
self, self,