雾聪
2023-10-10 75be3ceccdf02599c43c7e9888bfdb6ba561fc71
funasr/bin/ss_inference_launch.py
@@ -104,7 +104,7 @@
            ss_results = speech_separator(**batch)
            
            for spk in range(num_spks):
                sf.write(os.path.join(output_path, keys[0].replace('.wav', '_s'+str(spk+1)+'.wav')), ss_results[spk], sample_rate)
                sf.write(os.path.join(output_path, keys[0] + '_s' + str(spk+1)+'.wav'), ss_results[spk], sample_rate)
        torch.cuda.empty_cache()
        return ss_results