Gregniuki commited on
Commit
df9c347
·
verified ·
1 Parent(s): c593fa0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -210,7 +210,8 @@ def infer_batch(ref_audio, ref_text, gen_text_batches, exp_name, remove_silence,
210
  with torch.inference_mode():
211
  # Ensure all inputs are on the same device as ema_model
212
  audio = audio.to(ema_model.device) # Match ema_model's device
213
- final_text_list = [t.to(ema_model.device) if isinstance(t, torch.Tensor) else t for t in final_text_list] generated, _ = ema_model.sample(
 
214
  cond=audio,
215
  text=final_text_list,
216
  duration=duration,
 
210
  with torch.inference_mode():
211
  # Ensure all inputs are on the same device as ema_model
212
  audio = audio.to(ema_model.device) # Match ema_model's device
213
+ final_text_list = [t.to(ema_model.device) if isinstance(t, torch.Tensor) else t for t in final_text_list]
214
+ generated, _ = ema_model.sample(
215
  cond=audio,
216
  text=final_text_list,
217
  duration=duration,