add dtype property to the Generator

auffusion_pipeline.py

@@ -321,6 +321,14 @@ class Generator(torch.nn.Module):
 
         return wavs
 
+    @property
+    def dtype(self):
+        for param in self.parameters():
+            return param.dtype
+        for buffer in self.buffers():
+            return buffer.dtype
+        return torch.float32
+
 
 
 def normalize_spectrogram(