More flac support

2026-04-10 12:27:09 -06:00
parent 0b9c365acb
commit 9b4e2d1b0b
1 changed files with 1 additions and 1 deletions
--- a/toolkit/config_modules.py
+++ b/toolkit/config_modules.py
@@ -1200,7 +1200,7 @@ class GenerateImageConfig:
                )
            else:
                raise ValueError(f"Unsupported video format {self.output_ext}")
-        elif self.output_ext in ['wav', 'mp3']:
+        elif self.output_ext in ['wav', 'mp3', 'flac', 'ogg']:
            # save audio file
            audio_path = self.get_image_path(count, max_count)
            torchaudio.save(