diff --git a/crates/service/job/inference_job/src/job/job_types/tts/vall_e_x/vall_e_x_inference_command.rs b/crates/service/job/inference_job/src/job/job_types/tts/vall_e_x/vall_e_x_inference_command.rs index 9b43d9c4a4..53c6708ffd 100644 --- a/crates/service/job/inference_job/src/job/job_types/tts/vall_e_x/vall_e_x_inference_command.rs +++ b/crates/service/job/inference_job/src/job/job_types/tts/vall_e_x/vall_e_x_inference_command.rs @@ -44,7 +44,7 @@ static IGNORED_ENVIRONMENT_VARS : Lazy> = Lazy::new(|| { pub struct InferenceArgs> { /// --driven_audio: path to the input audio pub input_embedding_path: P, // name of the embedding.npz in the work dir - pub input_text: P, // name of the text + pub input_text: String, // name of the text /// --result_file: path to final file output pub output_file_name: P, // output file name in the output folder pub stderr_output_file: P @@ -187,7 +187,7 @@ impl VallEXInferenceCommand { // ===== Begin Python Args ===== command.push_str(" --text "); - command.push_str(&path_to_string(args.input_text)); + command.push_str(&format!("\"{}\"",&args.input_text)); command.push_str(" --prompt-name "); command.push_str(&path_to_string(args.input_embedding_path)); @@ -202,7 +202,7 @@ impl VallEXInferenceCommand { command.push_str(&path_to_string(Path::new("medium"))); command.push_str(" --whisper-folder-path "); - command.push_str(&path_to_string(Path::new("/tmp/downloads/zero_shot_tts/vall-e-x_1.0/medium.pt"))); + command.push_str(&path_to_string(Path::new("/tmp/downloads/zero_shot_tts/vall-e-x_1.0/whisper/"))); command.push_str(" --vocos-folder-path "); command.push_str(&path_to_string(Path::new("/tmp/downloads/zero_shot_tts/vall-e-x_1.0/vocos-encodec/"))); @@ -210,6 +210,9 @@ impl VallEXInferenceCommand { command.push_str(" --vallex-path "); command.push_str(&path_to_string(Path::new("/tmp/downloads/zero_shot_tts/vall-e-x_1.0/vallex-checkpoint.pt"))); + command.push_str(" --tmp-work-dir "); + command.push_str(&path_to_string(Path::new("/tmp/downloads/zero_shot_tts/"))); + // ===== End Python Args ===== if let Some(docker_options) = self.maybe_docker_options.as_ref() {