temporary fix for memory leak

parakeet/models/waveflow.py

@@ -483,7 +483,7 @@ class ConditionalWaveFlow(nn.LayerList):
         z, log_det_jacobian = self.decoder(audio, condition)
         return z, log_det_jacobian
     
-    @paddle.no_grad()
+    @paddle.fluid.dygraph.no_grad
     def synthesize(self, mel):
         condition = self.encoder(mel, trim_conv_artifact=True) #(B, C, T)
         batch_size, _, time_steps = condition.shape

parakeet/modules/losses.py

@@ -23,9 +23,3 @@ def masked_l1_loss(prediction, target, mask):
 def masked_softmax_with_cross_entropy(logits, label, mask, axis=-1):
     ce = F.softmax_with_cross_entropy(logits, label, axis=axis)
     return weighted_mean(ce, mask)
-
-
-
-
-    
-