jhtonyKoo commited on
Commit
7e88b50
1 Parent(s): 166943a

Update inference/style_transfer.py

Browse files
Files changed (1) hide show
  1. inference/style_transfer.py +2 -2
inference/style_transfer.py CHANGED
@@ -267,8 +267,8 @@ class Mixing_Style_Transfer_Inference:
267
  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')
268
  # remix
269
  fin_data_out_mix = sum(inst_outputs)
270
- fin_output_path = os.path.join(cur_out_dir, f"mixture_{output_name_tag}.wav"
271
- sf.write(fin_output_path), fin_data_out_mix.transpose(-1, -2), self.args.sample_rate, 'PCM_16')
272
 
273
  return fin_output_path
274
 
 
267
  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')
268
  # remix
269
  fin_data_out_mix = sum(inst_outputs)
270
+ fin_output_path = os.path.join(cur_out_dir, f"mixture_{output_name_tag}.wav")
271
+ sf.write(fin_output_path, fin_data_out_mix.transpose(-1, -2), self.args.sample_rate, 'PCM_16')
272
 
273
  return fin_output_path
274