Update f5_tts/model/cfm.py

f5_tts/model/cfm.py

@@ -161,7 +161,7 @@ class CFM(nn.Module):
 
         def fn(t, x):
             # at each step, conditioning is fixed
-            step_cond = torch.where(cond_mask, cond, torch.zeros_like(cond))
+            # step_cond = torch.where(cond_mask, cond, torch.zeros_like(cond))
 
             # predict flow
             pred = self.transformer(