diff --git a/crates/service/job/download_job/download-job.env b/crates/service/job/download_job/download-job.env index 70d8065342..36f9a5235b 100644 --- a/crates/service/job/download_job/download-job.env +++ b/crates/service/job/download_job/download-job.env @@ -31,4 +31,5 @@ HIFIGAN_MODEL_CHECK_MAYBE_DOCKER_IMAGE=5642d0fd7fc1 VITS_MODEL_CHECK_ROOT_DIRECTORY="/vits" VITS_MODEL_CHECK_COMMAND="export_ts.py" #VITS_MODEL_CHECK_MAYBE_VENV_COMMAND="source python/bin/activate" +VITS_MODEL_CHECK_MAYBE_PYTHON_INTERPRETER="python3" VITS_MODEL_CHECK_MAYBE_DOCKER_IMAGE="1d5b8788ac28" diff --git a/crates/service/job/download_job/src/job_types/tts/vits/process_vits_model.rs b/crates/service/job/download_job/src/job_types/tts/vits/process_vits_model.rs index a989c68012..43ed30f64d 100644 --- a/crates/service/job/download_job/src/job_types/tts/vits/process_vits_model.rs +++ b/crates/service/job/download_job/src/job_types/tts/vits/process_vits_model.rs @@ -37,7 +37,7 @@ pub async fn process_vits_model<'a, 'b>( let original_model_file_path = PathBuf::from(download_filename.clone()); // NB: We'll be creating the traced model in the "check" step and uploading it to GDrive along with the original. - let traced_model_file_path = PathBuf::from(format!("{:?}_traced", original_model_file_path)); + let traced_model_file_path = PathBuf::from(format!("{:?}_traced", original_model_file_path.as_os_str())); let config_path = PathBuf::from("configs/ljs_li44_tmbert_nmp_s1_arpa.json"); // TODO: This could be variable. diff --git a/crates/service/job/download_job/src/main.rs b/crates/service/job/download_job/src/main.rs index bee7e045d3..b05165783d 100644 --- a/crates/service/job/download_job/src/main.rs +++ b/crates/service/job/download_job/src/main.rs @@ -266,6 +266,9 @@ async fn main() -> AnyhowResult<()> { let maybe_venv_command = easyenv::get_env_string_optional( "VITS_MODEL_CHECK_MAYBE_VENV_COMMAND"); + let maybe_python_interpreter = easyenv::get_env_string_optional( + "VITS_MODEL_CHECK_MAYBE_PYTHON_INTERPRETER"); + let maybe_docker_options = easyenv::get_env_string_optional( "VITS_MODEL_CHECK_MAYBE_DOCKER_IMAGE") .map(|image_name| { @@ -279,7 +282,7 @@ async fn main() -> AnyhowResult<()> { VitsModelCheckCommand::new( root_directory, check_script, - None, + maybe_python_interpreter.as_deref(), maybe_venv_command.as_deref(), maybe_docker_options, )