Gregniuki commited on
Commit
915f88b
·
verified ·
1 Parent(s): d1df1d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- 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
 
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