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

2025-06-04 08:46:37 +08:00
parent 0440ca24bb 9c5c92946d
commit d612523f50
1 changed files with 5 additions and 2 deletions
@@ -1393,9 +1393,12 @@ class ACEStepPipeline:
                output_path_wav = save_path

        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(
-            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