Spaces:
Running
on
Zero
Running
on
Zero
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
|