from local

This commit is contained in:
dohe0342 2023-04-11 16:32:16 +09:00
parent 796be81be8
commit 1ebc50a4ec
3 changed files with 1 additions and 1 deletions

View File

@ -79,7 +79,7 @@ class Transducer(nn.Module):
self.prompt = None
if prompt:
self.prompt = torch.randn((50, 512), requires_grad=True).to('cuda')
self.prompt = torch.randn((50, 512), requires_grad=True)
def forward(
self,