diff --git a/src/AceStepWorker.cpp b/src/AceStepWorker.cpp
index fc999f2..1528805 100644
--- a/src/AceStepWorker.cpp
+++ b/src/AceStepWorker.cpp
@@ -181,6 +181,7 @@ void AceStep::qwenProcFinished(int code, QProcess::ExitStatus status)
 	ditVaeArgs << "--text-encoder"<<request.textEncoderModelPath;
 	ditVaeArgs << "--dit"<<request.ditModelPath;
 	ditVaeArgs << "--vae"<<request.vaeModelPath;
+	ditVaeArgs << "--wav";
 
 	progressUpdate(60);