Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -225,7 +225,7 @@ def infer_batch(ref_audio, ref_text, gen_text_batches, exp_name, remove_silence,
|
|
| 225 |
generated_mel_spec = rearrange(generated, "1 n d -> 1 d n")
|
| 226 |
|
| 227 |
# Convert to appropriate dtype and device
|
| 228 |
-
|
| 229 |
generated_wave = vocos.decode(generated_mel_spec)
|
| 230 |
|
| 231 |
# Adjust wave RMS if needed
|
|
|
|
| 225 |
generated_mel_spec = rearrange(generated, "1 n d -> 1 d n")
|
| 226 |
|
| 227 |
# Convert to appropriate dtype and device
|
| 228 |
+
# generated_mel_spec = generated_mel_spec.to(dtype=torch.float16, device=vocos.device) # Ensure device matches vocos
|
| 229 |
generated_wave = vocos.decode(generated_mel_spec)
|
| 230 |
|
| 231 |
# Adjust wave RMS if needed
|