fix errors

This commit is contained in:
salt committed 2023-12-12 15:12:34 -08:00
1 parent 43030402ca
commit 14432cce83
7 files changed
+41 -25

No files matched your search

@@ -16,6 +16,7 @@ use crate::job::job_loop::process_single_job_success_case::ProcessSingleJobSucce
use crate::job::job_types::lipsync::process_single_lipsync_job::process_single_lipsync_job;
use crate::job::job_types::tts::process_single_tts_job::process_single_tts_job;
use crate::job::job_types::vc::process_single_vc_job::process_single_vc_job;
use crate::job::job_types::videofilter::process_single_vf_job::process_single_rr_job;
use crate::job_dependencies::JobDependencies;
pub async fn process_single_job(
@@ -183,7 +184,7 @@ async fn do_process_single_job(
process_single_vc_job(job_dependencies, job).await?
}
InferenceCategory::RerenderAVideo => {
process_single_rerender_job(job_dependencies, job).await?
process_single_rr_job(job_dependencies, job).await?
}
};
@@ -1,2 +1,2 @@
pub(crate) mod rerender_a_video;
mod process_single_vf_job;
pub(crate) mod process_single_vf_job;
@@ -1,19 +1,13 @@
use std::time::Duration;
use anyhow::anyhow;
use log::{error, info};
use enums::by_table::voice_conversion_models::voice_conversion_model_type::VoiceConversionModelType;
use mysql_queries::queries::generic_inference::job::list_available_generic_inference_jobs::AvailableInferenceJob;
use mysql_queries::queries::media_uploads::get_media_upload_for_inference::get_media_upload_for_inference;
use mysql_queries::queries::voice_conversion::inference::get_voice_conversion_model_for_inference::get_voice_conversion_model_for_inference;
use tokens::tokens::media_uploads::MediaUploadToken;
use crate::job::job_loop::job_success_result::JobSuccessResult;
use crate::job::job_loop::process_single_job_error::ProcessSingleJobError;
use crate::job::job_types::vc::{rvc_v2, so_vits_svc};
use crate::job::job_types::vc::rvc_v2::process_job::RvcV2ProcessJobArgs;
use crate::job::job_types::vc::so_vits_svc::process_job::SoVitsSvcProcessJobArgs;
use crate::job::job_types::videofilter::rerender_a_video;
use crate::job_dependencies::JobDependencies;
pub async fn process_single_rr_job(job_dependencies: &JobDependencies, job: &AvailableInferenceJob) -> Result<JobSuccessResult, ProcessSingleJobError> {
@@ -47,17 +41,24 @@ pub async fn process_single_rr_job(job_dependencies: &JobDependencies, job: &Ava
let job_success_result = match vc_model.model_type {
VoiceConversionModelType::RvcV2 => {
rvc_v2::process_job::process_job(RvcV2ProcessJobArgs {
job_dependencies,
job,
vc_model: &vc_model,
media_upload_token: &media_upload_token,
media_upload: &media_upload,
}).await?
// let job_success_result = match vc_model.model_type {
// VoiceConversionModelType::RvcV2 => {
// rvc_v2::process_job::process_job(RvcV2ProcessJobArgs {
// job_dependencies,
// job,
// vc_model: &vc_model,
// media_upload_token: &media_upload_token,
// media_upload: &media_upload,
// }).await?
// }
// };
let job_success_result = rerender_a_video::process_job::process_job(
rerender_a_video::process_job::RerenderProcessJobArgs {
job_dependencies,
job
}
};
).await?;
Ok(job_success_result)
}
@@ -1,4 +1,4 @@
mod process_jobs;
pub(crate) mod process_job;
mod validate_job;
mod download_video_file;
mod rerender_inference_command;
@@ -77,9 +77,9 @@ impl RerenderInferenceCommand {
maybe_execution_timeout: Option<Duration>,
) -> Self {
Self {
rerender_root_code_directory: rerender_root_code_directory.as_ref().to_path_buf(),
rerender_root_code_directory: rerender_root_code_directory.clone(),
executable_or_command,
config_path: config_path.as_ref().to_path_buf(),
config_path: config_path.clone(),
maybe_virtual_env_activation_command: maybe_virtual_env_activation_command.map(|s| s.to_string()),
maybe_docker_options,
maybe_execution_timeout,
@@ -93,6 +93,20 @@ impl RerenderInferenceCommand {
let config_path = easyenv::get_env_pathbuf_required(
"RERENDER_INFERENCE_CONFIG_PATH")?;
let executable_or_command = match easyenv::get_env_string_optional(
"RERENDER_INFERENCE_EXECUTABLE_OR_COMMAND") {
None => {
return Err(anyhow!("RERENDER_INFERENCE_EXECUTABLE_OR_COMMAND is required"));
}
Some(executable_or_command) => {
if executable_or_command.contains(" ") {
ExecutableOrCommand::Command(executable_or_command)
} else {
ExecutableOrCommand::Executable(PathBuf::from(executable_or_command))
}
}
};
let maybe_virtual_env_activation_command = easyenv::get_env_string_optional(
"RERENDER_INFERENCE_MAYBE_VENV_COMMAND");
@@ -192,7 +206,7 @@ impl RerenderInferenceCommand {
let mut config = PopenConfig::default();
info!("stderr will be written to file: {:?}", args.stderr_output_file.as_ref());
info!("stderr will be written to file: {:?}", args.stderr_output_file.as_os_str());
let stderr_file = File::create(&args.stderr_output_file)?;
config.stderr = Redirection::File(stderr_file);
@@ -1,8 +1,8 @@
use actix_web::HttpRequest;
use mockall::automock;
// use mockall::automock;
/// Allows us to purge session caches that might keep stale premium plan/billing data around
#[cfg_attr(test, automock)]
// #[cfg_attr(test, automock)]
pub trait InternalSessionCachePurge {
/// Best effort attempt to delete
/// Fails silently.