Fix hifigan (softvc) model check

This commit is contained in:
Brandon Thomas
2023-01-11 00:33:39 -05:00
parent 51350e6e5c
commit c2f4c2c057
3 changed files with 5 additions and 3 deletions
+2 -2
View File
@@ -8,7 +8,7 @@ WEB_DOWNLOADER_MAYBE_DOCKER_IMAGE=d73f28ce3ff6
SOFTVC_MODEL_CHECK_ROOT_DIRECTORY=/
SOFTVC_MODEL_CHECK_MAYBE_VENV_COMMAND="source python/bin/activate"
SOFTVC_MODEL_CHECK_MAYBE_DOCKER_IMAGE=036b23daa586
SOFTVC_MODEL_CHECK_MAYBE_DOCKER_IMAGE=a79106991ce3
TACOTRON_MODEL_CHECK_ROOT_DIRECTORY=/models/tts
TACOTRON_MODEL_CHECK_MAYBE_VENV_COMMAND="source python/bin/activate"
@@ -17,7 +17,7 @@ TACOTRON_MODEL_CHECK_MAYBE_DOCKER_IMAGE=5642d0fd7fc1
HIFIGAN_SOFTVC_MODEL_CHECK_ROOT_DIRECTORY=/
HIFIGAN_SOFTVC_MODEL_CHECK_MAYBE_VENV_COMMAND="source python/bin/activate"
HIFIGAN_SOFTVC_MODEL_CHECK_MAYBE_DOCKER_IMAGE=036b23daa586
HIFIGAN_SOFTVC_MODEL_CHECK_MAYBE_DOCKER_IMAGE=a79106991ce3
HIFIGAN_MODEL_CHECK_ROOT_DIRECTORY=/models/tts
HIFIGAN_MODEL_CHECK_MAYBE_VENV_COMMAND="source python/bin/activate"
@@ -28,6 +28,8 @@ pub async fn process_single_job(job_state: &JobState, job: &AvailableDownloadJob
return Ok(())
}
info!("Beginning work on {:?} = {} | {}", job.download_type, job.download_job_token, job.download_url);
// ==================== SETUP TEMP DIRS ==================== //
let temp_dir = format!("temp_{}", job.id.0);
@@ -35,7 +35,7 @@ pub async fn process_hifigan_softvc_vocoder<'a, 'b>(
let output_metadata_fs_path = temp_dir.path().join("metadata.json");
let model_check_result = job_state.sidecar_configs.hifigan_model_check_command.execute(
let model_check_result = job_state.sidecar_configs.hifigan_softvc_model_check_command.execute(
&file_path,
&output_metadata_fs_path
);