Spaces:

mehdi999
/

pardi-speech

Running on Zero

mehdi999 commited on Oct 29

Commit

bb32e4f

1 Parent(s): 80ec7ec

fix(GLA): provide safe past_key_values with default conv_state=(None,None,None)

Files changed (1) hide show

tts/model/simple_gla.py CHANGED Viewed

@@ -72,7 +72,7 @@ class SimpleGLABlock(nn.Module):
         + x
       )
       x = self.cmix(self.norm2(x)) + x
-    return x
 class DecoderBlockWithOptionalCrossAttention(nn.Module):

         + x
       )
       x = self.cmix(self.norm2(x)) + x
+      return x
 class DecoderBlockWithOptionalCrossAttention(nn.Module):