futurespyhi commited on
Commit
af9b37e
·
1 Parent(s): 6d4b73b

modify the false logic of vocal and instru matching

Browse files
Files changed (1) hide show
  1. YuE/inference/infer.py +2 -2
YuE/inference/infer.py CHANGED
@@ -440,8 +440,8 @@ for inst_path in tracks:
440
  continue
441
  # mix
442
  recons_mix = os.path.join(recons_mix_dir, os.path.basename(inst_path).replace('_itrack', '_mixed'))
443
- vocal_stem, sr = sf.read(inst_path)
444
- instrumental_stem, _ = sf.read(vocal_path)
445
  mix_stem = (vocal_stem + instrumental_stem) / 1
446
  sf.write(recons_mix, mix_stem, sr)
447
  except Exception as e:
 
440
  continue
441
  # mix
442
  recons_mix = os.path.join(recons_mix_dir, os.path.basename(inst_path).replace('_itrack', '_mixed'))
443
+ vocal_stem, sr = sf.read(vocal_path)
444
+ instrumental_stem, _ = sf.read(inst_path)
445
  mix_stem = (vocal_stem + instrumental_stem) / 1
446
  sf.write(recons_mix, mix_stem, sr)
447
  except Exception as e: