Outlined the things needed to be done in process_job

This commit is contained in:
Michael Chung committed 2023-10-06 00:49:42 -04:00
1 parent ba76acec35
commit 195a2eeb9c
3 files changed
+18 -3

No files matched your search

@@ -10,8 +10,11 @@ use crate::job::job_loop::process_single_job_error::ProcessSingleJobError;
use crate::job::job_types::tts::{tacotron2_v2_early_fakeyou, vits};
use crate::job::job_types::tts::tacotron2_v2_early_fakeyou::process_job::ProcessJobArgs;
use crate::job::job_types::tts::vits::process_job::VitsProcessJobArgs;
use crate::job::job_types::tts::vall_e_x::process_job::VALLEXProcessJobArgs;
use crate::job_dependencies::JobDependencies;
use super::vall_e_x;
pub async fn process_single_tts_job(job_dependencies: &JobDependencies, job: &AvailableInferenceJob) -> Result<JobSuccessResult, ProcessSingleJobError> {
let tts_model_token = match job.maybe_model_token.as_deref() {
@@ -60,7 +63,16 @@ pub async fn process_single_tts_job(job_dependencies: &JobDependencies, job: &Av
// Look for the Zeroshot job type
let job_success_result = match job.maybe_model_type {
Some(InferenceModelType::VallEX) => {
vall_e_x::process_job::process_job(VALLEXProcessJobArgs {
job_dependencies,
job,
}).await?
}
Some(model_type) => return Err(ProcessSingleJobError::Other(anyhow!("wrong model type: {:?}", model_type))),
None => return Err(ProcessSingleJobError::Other(anyhow!("no model type in record"))),
};
// let these run anyways ?
@@ -1,2 +1,5 @@
pub mod model_downloaders;
pub mod vall_e_x_inference_command;
pub mod vall_e_x_inference_command;
pub mod process_job;
pub mod validate_job;
pub mod download_embedding_file;
@@ -25,7 +25,7 @@ use crate::job::job_loop::process_single_job_error::ProcessSingleJobError;
use crate::job_dependencies::JobDependencies;
pub struct VALL_E_XProcessJobArgs<'a> {
pub struct VALLEXProcessJobArgs<'a> {
pub job_dependencies: &'a JobDependencies,
pub job: &'a AvailableInferenceJob,
}