Gregniuki commited on
Commit
66e7edf
·
verified ·
1 Parent(s): dd7f07b

Update app.py

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