From 081307d735d5b447261bdd0f0a02adbb916e3f07 Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Fri, 24 Jan 2025 01:13:53 -0500 Subject: [PATCH] debug sleep, width height --- .../process_single_studio_gen2_job.rs | 21 ++++++++++++------- 1 file changed, 14 insertions(+), 7 deletions(-) diff --git a/crates/service/job/inference_job/src/job/job_types/studio_gen2/process_single_studio_gen2_job.rs b/crates/service/job/inference_job/src/job/job_types/studio_gen2/process_single_studio_gen2_job.rs index b796ab0694..c4340bd612 100644 --- a/crates/service/job/inference_job/src/job/job_types/studio_gen2/process_single_studio_gen2_job.rs +++ b/crates/service/job/inference_job/src/job/job_types/studio_gen2/process_single_studio_gen2_job.rs @@ -41,6 +41,7 @@ use tokens::tokens::media_files::MediaFileToken; use tokio::sync::oneshot; use tokio::sync::oneshot::channel; use filesys::create_dir_all_if_missing::create_dir_all_if_missing; +use mysql_queries::payloads::generic_inference_args::inner_payloads::studio_gen2_payload::StudioGen2Payload; use videos::ffprobe_get_dimensions::ffprobe_get_dimensions; use crate::job::job_types::studio_gen2::stable_animator_command::InferenceArgs; use crate::job::job_types::studio_gen2::validate_and_save_results::{validate_and_save_results, SaveResultsArgs}; @@ -249,6 +250,8 @@ pub async fn process_single_studio_gen2_job( warn!("Captured stderr output: {}", contents); } + maybe_debug_sleep(studio_args).await; + // NB: Forcing generic type to `&Path` with turbofish safe_delete_possible_files_and_directories::<&Path>(&[ Some(work_paths.input_dir.path()), @@ -296,6 +299,8 @@ pub async fn process_single_studio_gen2_job( Err(err) => { error!("Error validating and saving results: {:?}", err); + maybe_debug_sleep(studio_args).await; + // NB: Forcing generic type to `&Path` with turbofish safe_delete_possible_files_and_directories::<&Path>(&[ Some(work_paths.input_dir.path()), @@ -306,15 +311,10 @@ pub async fn process_single_studio_gen2_job( } }; - // ==================== (OPTIONAL) DEBUG SLEEP ==================== // - - if let Some(sleep_millis) = studio_args.after_job_debug_sleep_millis { - info!("Sleeping for millis: {sleep_millis}"); - tokio::time::sleep(Duration::from_millis(sleep_millis)).await; - } - // ==================== CLEANUP/ DELETE TEMP FILES ==================== // + maybe_debug_sleep(studio_args).await; + info!("Cleaning up temporary files..."); // NB: Forcing generic type to `&Path` with turbofish @@ -342,3 +342,10 @@ pub async fn process_single_studio_gen2_job( inference_duration, }) } + +async fn maybe_debug_sleep(args: &StudioGen2Payload) { + if let Some(sleep_millis) = args.after_job_debug_sleep_millis { + info!("Debug sleeping for millis: {sleep_millis}"); + tokio::time::sleep(Duration::from_millis(sleep_millis)).await; + } +}