From 7c553ffcee7a6727c35991bf03382509950b2369 Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Fri, 20 Oct 2023 07:00:09 -0400
Subject: [PATCH] fix rvc jobs

---
 .../src/job/job_types/vc/rvc_v2/process_job.rs             | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/crates/service/job/inference_job/src/job/job_types/vc/rvc_v2/process_job.rs b/crates/service/job/inference_job/src/job/job_types/vc/rvc_v2/process_job.rs
index f1c0c0d7c7..bd81ae852f 100644
--- a/crates/service/job/inference_job/src/job/job_types/vc/rvc_v2/process_job.rs
+++ b/crates/service/job/inference_job/src/job/job_types/vc/rvc_v2/process_job.rs
@@ -3,10 +3,9 @@ use std::time::Instant;
 
 use anyhow::anyhow;
 use log::{error, info, warn};
-use buckets::public::media_files::original_file::MediaFileBucketPath;
 
+use buckets::public::media_files::original_file::MediaFileBucketPath;
 use buckets::public::media_uploads::original_file::MediaUploadOriginalFilePath;
-use buckets::public::voice_conversion_results::original_file::VoiceConversionResultOriginalFilePath;
 use container_common::filesystem::check_file_exists::check_file_exists;
 use container_common::filesystem::safe_delete_temp_directory::safe_delete_temp_directory;
 use container_common::filesystem::safe_delete_temp_file::safe_delete_temp_file;
@@ -362,7 +361,7 @@ pub async fn process_job(args: RvcV2ProcessJobArgs<'_>) -> Result<JobSuccessResu
   let (inference_result_token, id) = insert_media_file_from_voice_conversion(InsertMediaFileArgs {
     pool: &args.job_dependencies.mysql_pool,
     job: &job,
-    voice_conversion_type: VoiceConversionModelType::SoVitsSvc,
+    voice_conversion_type: VoiceConversionModelType::RvcV2,
     maybe_mime_type: maybe_mimetype.as_deref(),
     file_size_bytes: file_metadata.file_size_bytes,
     duration_millis: file_metadata.duration_millis.unwrap_or(0),
@@ -401,7 +400,7 @@ pub async fn process_job(args: RvcV2ProcessJobArgs<'_>) -> Result<JobSuccessResu
 
   Ok(JobSuccessResult {
     maybe_result_entity: Some(ResultEntity {
-      entity_type: InferenceResultType::VoiceConversion,
+      entity_type: InferenceResultType::MediaFile,
       entity_token: inference_result_token.to_string(),
     }),
     inference_duration,