mirror of
https://github.com/storytold/artcraft.git
synced 2026-11-03 00:47:53 +00:00
Outlined the things needed to be done in process_job
This commit is contained in:
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,
|
||||
}
|
||||
|
||||
Reference in new issue
Block a user