oops

2023-10-11 20:49:54 -05:00 · 2023-10-11 20:49:54 -05:00 · 3af19d79fd
commit 3af19d79fd
parent 8740cdefc6
1 changed files with 4 additions and 1 deletions
--- a/vall_e/models/base.py
+++ b/vall_e/models/base.py
@ -191,7 +191,10 @@ class Base(nn.Module):
 			cat = torch.cat
 		else:
 			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__(
 		self,