diff --git a/toolkit/config_modules.py b/toolkit/config_modules.py
index a5bdeab..57edae0 100644
--- 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(