daniel543x/artcraft
1
0 Fork 0
mirror of https://github.com/storytold/artcraft.git synced 2026-11-13 05:12:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

logging

Browse Source
This commit is contained in:
Brandon Thomas committed 2023-04-04 14:25:50 -07:00
1 parent abaaaf2fdb
commit a83e5ff224
1 file changed
+8 -5
Download Patch File Download Diff File Expand all files Collapse all files
crates/service/job/download_job/src/job_types/tts/vits/process_vits_model.rs
+8 -5
View File
@@ -10,7 +10,7 @@ use enums::common::visibility::Visibility;
use errors::AnyhowResult;
use hashing::sha256::sha256_hash_file::sha256_hash_file;
use jobs_common::redis_job_status_logger::RedisJobStatusLogger;
use log::{info, warn};
use log::{error, info, warn};
use mysql_queries::queries::generic_download::job::list_available_generic_download_jobs::AvailableDownloadJob;
use mysql_queries::queries::tts::tts_models::insert_tts_model_from_download_job::insert_tts_model_from_download_job;
use mysql_queries::queries::tts::tts_models::insert_tts_model_from_download_job;
@@ -89,11 +89,12 @@ pub async fn process_vits_model<'a, 'b>(
redis_logger.log_status("uploading VITS TTS model")?;
if let Err(e) = job_state.bucket_client.upload_filename(&model_bucket_path, &original_model_file_path).await {
if let Err(err) = job_state.bucket_client.upload_filename(&model_bucket_path, &original_model_file_path).await {
error!("Problem uploading original model: {:?}", err);
safe_delete_temp_file(&original_model_file_path);
safe_delete_temp_file(&traced_model_file_path);
safe_delete_temp_directory(&temp_dir);
return Err(e);
return Err(err);
}
// ==================== UPLOAD TRACED MODEL FILE ==================== //
@@ -106,16 +107,18 @@ pub async fn process_vits_model<'a, 'b>(
redis_logger.log_status("uploading VITS TTS (traced) model")?;
if let Err(e) = job_state.bucket_client.upload_filename(&traced_model_bucket_path, &traced_model_bucket_path).await {
if let Err(err) = job_state.bucket_client.upload_filename(&traced_model_bucket_path, &traced_model_bucket_path).await {
error!("Problem uploading traced model: {:?}", err);
safe_delete_temp_file(&original_model_file_path);
safe_delete_temp_file(&traced_model_file_path);
safe_delete_temp_directory(&temp_dir);
return Err(e);
return Err(err);
}
// ==================== DELETE DOWNLOADED FILE ==================== //
// NB: We should be using a tempdir, but to make absolutely certain we don't overflow the disk...
info!("Done uploading; deleting temporary files and paths...");
safe_delete_temp_file(&original_model_file_path);
safe_delete_temp_file(&traced_model_file_path);
safe_delete_temp_directory(&temp_dir);