Politrees commited on
Commit
566686c
1 Parent(s): 72c5bf9

Update src/infer_pack/predictor/RMVPE.py

Browse files
Files changed (1) hide show
  1. src/infer_pack/predictor/RMVPE.py +1 -0
src/infer_pack/predictor/RMVPE.py CHANGED
@@ -352,6 +352,7 @@ class RMVPE:
352
  def mel2hidden(self, mel):
353
  with torch.no_grad():
354
  n_frames = mel.shape[-1]
 
355
  mel = F.pad(
356
  mel, (0, 32 * ((n_frames - 1) // 32 + 1) - n_frames), mode="reflect"
357
  )
 
352
  def mel2hidden(self, mel):
353
  with torch.no_grad():
354
  n_frames = mel.shape[-1]
355
+ mel = mel.float() # Приведение к float32 перед паддингом
356
  mel = F.pad(
357
  mel, (0, 32 * ((n_frames - 1) // 32 + 1) - n_frames), mode="reflect"
358
  )