mirror of
https://github.com/storytold/artcraft.git
synced 2026-11-05 17:58:55 +00:00
fix errors
This commit is contained in:
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;
|
||||
+18
-17
@@ -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
-1
@@ -1,4 +1,4 @@
|
||||
mod process_jobs;
|
||||
pub(crate) mod process_job;
|
||||
mod validate_job;
|
||||
mod download_video_file;
|
||||
mod rerender_inference_command;
|
||||
|
||||
File renamed without changes.
+17
-3
@@ -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.
|
||||
|
||||
Reference in new issue
Block a user