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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -240,8 +240,12 @@ def infer_batch(ref_audio, ref_text, gen_text_batches, exp_name, remove_silence,
240
  generated_wave = generated_wave.squeeze().cpu().numpy()
241
 
242
  generated_waves.append(generated_wave)
243
- spectrograms.append(generated_mel_spec[0].cpu().numpy())
 
 
244
 
 
 
245
  # Combine all generated waves with cross-fading
246
  if cross_fade_duration <= 0:
247
  # Simply concatenate
 
240
  generated_wave = generated_wave.squeeze().cpu().numpy()
241
 
242
  generated_waves.append(generated_wave)
243
+ # spectrograms.append(generated_mel_spec[0].cpu().numpy())
244
+ # Ensure generated_mel_spec is in a compatible dtype (e.g., float32) before passing it to numpy
245
+ generated_mel_spec = generated_mel_spec.to(dtype=torch.float32) # Convert to float32 if it's in bfloat16
246
 
247
+ # Proceed with the rest of your operations
248
+ spectrograms.append(generated_mel_spec[0].cpu().numpy())
249
  # Combine all generated waves with cross-fading
250
  if cross_fade_duration <= 0:
251
  # Simply concatenate