mrleo1nid/ACE-Step
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #257 from SD-inst/ogg-fix

Browse Source
This commit is contained in:
Gong Junmin
2025-06-04 08:46:37 +08:00
committed by GitHub
parent 0440ca24bb 9c5c92946d
commit d612523f50
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
acestep/pipeline_ace_step.py
+5 -2
View File
@@ -1393,9 +1393,12 @@ class ACEStepPipeline:
output_path_wav = save_path output_path_wav = save_path
target_wav = target_wav.float() target_wav = target_wav.float()
logger.info(f"Saving audio to {output_path_wav}") backend = "soundfile"
if format == "ogg":
backend = "sox"
logger.info(f"Saving audio to {output_path_wav} using backend {backend}")
torchaudio.save( torchaudio.save(
output_path_wav, target_wav, sample_rate=sample_rate, format=format output_path_wav, target_wav, sample_rate=sample_rate, format=format, backend=backend
) )
return output_path_wav return output_path_wav