mrq/vall-e
5
9
Fork 0
Code Issues 8 Pull Requests Packages Projects Releases Wiki Activity

oops

Browse Source
This commit is contained in:
mrq 2023-10-11 20:49:54 -05:00
parent 8740cdefc6
commit 3af19d79fd
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
vall_e/models/base.py
View File

@ -191,7 +191,10 @@ class Base(nn.Module):
cat = torch.cat cat = torch.cat
else: else:
cat = partial(_join, sep=sep) cat = partial(_join, sep=sep)
return [*map(cat, zip([x for x in l if x is not None]))]
l = [ x for x in l if x is not None ]
return [*map(cat, zip(*l))]
def __init__( def __init__(
self, self,