jhtonyKoo commited on
Commit
7df90aa
1 Parent(s): 7295413

Update inference/style_transfer.py

Browse files
Files changed (1) hide show
  1. inference/style_transfer.py +2 -2
inference/style_transfer.py CHANGED
@@ -264,8 +264,8 @@ class Mixing_Style_Transfer_Inference:
264
  sf.write(os.path.join(cur_out_dir, f"{cur_inst_name}_{output_name_tag}.wav"), fin_data_out_inst.transpose(-1, -2), self.args.sample_rate, 'PCM_16')
265
  # remix
266
  fin_data_out_mix = sum(inst_outputs)
267
- fin_output_path = os.path.join(cur_out_dir, f"mixture_{output_name_tag}.wav"
268
- sf.write(fin_output_path), fin_data_out_mix.transpose(-1, -2), self.args.sample_rate, 'PCM_16')
269
 
270
  return fin_output_path
271
 
 
264
  sf.write(os.path.join(cur_out_dir, f"{cur_inst_name}_{output_name_tag}.wav"), fin_data_out_inst.transpose(-1, -2), self.args.sample_rate, 'PCM_16')
265
  # remix
266
  fin_data_out_mix = sum(inst_outputs)
267
+ fin_output_path = os.path.join(cur_out_dir, f"mixture_{output_name_tag}.wav")
268
+ sf.write(fin_output_path, fin_data_out_mix.transpose(-1, -2), self.args.sample_rate, 'PCM_16')
269
 
270
  return fin_output_path
271