daniel543x/artcraft
1
0 Fork 0
mirror of https://github.com/storytold/artcraft.git synced 2026-11-04 17:38:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

auto-configure jobs with job type

Browse Source
This commit is contained in:
Brandon Thomas committed 2024-09-05 00:55:27 -04:00
1 parent c65bcef4c0
commit 0dbc056546
1 file changed
+12 -4
Download Patch File Download Diff File Expand all files Collapse all files
crates/service/job/inference_job/src/state/job_specific_dependencies.rs
+12 -4
View File
@@ -78,17 +78,23 @@ impl JobSpecificDependencies {
maybe_rvc_v2_dependencies = Some(RvcV2Dependencies::setup()?);
}
if scoped_model_type_execution.can_run_job(InferenceModelType::SadTalker) {
if scoped_model_type_execution.can_run_job(InferenceModelType::SadTalker)
|| scoped_job_type_execution.can_run_job(InferenceJobType::SadTalker)
{
print_with_space("Setting SadTalker dependencies...");
maybe_sad_talker_dependencies = Some(SadTalkerDependencies::setup()?);
}
if scoped_model_type_execution.can_run_job(InferenceModelType::SoVitsSvc) {
if scoped_model_type_execution.can_run_job(InferenceModelType::SoVitsSvc)
|| scoped_job_type_execution.can_run_job(InferenceJobType::SoVitsSvc)
{
print_with_space("Setting SVC dependencies...");
maybe_svc_dependencies = Some(SvcDependencies::setup()?);
}
if scoped_model_type_execution.can_run_job(InferenceModelType::Tacotron2) {
if scoped_model_type_execution.can_run_job(InferenceModelType::Tacotron2)
|| scoped_job_type_execution.can_run_job(InferenceJobType::Tacotron2)
{
print_with_space("Setting Tacotron2 dependencies...");
maybe_tacotron2_dependencies = Some(Tacotron2Dependencies::setup()?);
}
@@ -98,7 +104,9 @@ impl JobSpecificDependencies {
maybe_vall_e_x_dependencies = Some(VallExDependencies::setup()?);
}
if scoped_model_type_execution.can_run_job(InferenceModelType::StyleTTS2) {
if scoped_model_type_execution.can_run_job(InferenceModelType::StyleTTS2)
|| scoped_job_type_execution.can_run_job(InferenceJobType::StyleTTS2)
{
print_with_space("Setting StyleTTS2 dependencies...");
maybe_styletts2_dependencies = Some(StyleTTS2Dependencies::setup()?);
}