minor fix.

Author: SWivid

model/cfm.py

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