Update app.py

app.py

@@ -219,7 +219,7 @@ def infer_batch(ref_audio, ref_text, gen_text_batches, exp_name, remove_silence,
 
         generated = generated[:, ref_audio_len:, :]
         generated_mel_spec = rearrange(generated, "1 n d -> 1 d n")
-       # generated_mel_spec = generated_mel_spec.to(dtype=torch.bfloat16)  # Convert to bfloat16
+        generated_mel_spec = generated_mel_spec.to(dtype=torch.float16)  # Convert to bfloat16
         generated_wave = vocos.decode(generated_mel_spec.cpu())
         if rms < target_rms:
             generated_wave = generated_wave * rms / target_rms