Update inference/style_transfer.py
Browse files
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
|
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 |
|