Sync from GitHub repo

This Space is synced from the GitHub repo: https://github.com/SWivid/F5-TTS. Please submit contributions to the Space there

model/cfm.py

@@ -96,7 +96,7 @@ class CFM(nn.Module):
     ):
         self.eval()
 
-        if cond.device == torch.device('cuda'):
+        if next(self.parameters()).dtype == torch.float16:
             cond = cond.half()
 
         # raw wave