Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -231,7 +231,7 @@ def infer_batch(ref_audio, ref_text, gen_text_batches, exp_name, remove_silence,
|
|
231 |
|
232 |
generated = generated[:, ref_audio_len:, :]
|
233 |
generated_mel_spec = rearrange(generated, "1 n d -> 1 d n")
|
234 |
-
|
235 |
generated_wave = vocos.decode(generated_mel_spec.cpu())
|
236 |
if rms < target_rms:
|
237 |
generated_wave = generated_wave * rms / target_rms
|
|
|
231 |
|
232 |
generated = generated[:, ref_audio_len:, :]
|
233 |
generated_mel_spec = rearrange(generated, "1 n d -> 1 d n")
|
234 |
+
# generated_mel_spec = generated_mel_spec.to(dtype=torch.bfloat16) # Convert to bfloat16
|
235 |
generated_wave = vocos.decode(generated_mel_spec.cpu())
|
236 |
if rms < target_rms:
|
237 |
generated_wave = generated_wave * rms / target_rms
|