queue stats for inference-job's tacotron2

This commit is contained in:
Brandon Thomas committed 2023-11-28 07:27:26 -05:00
1 parent 8176f0d6bd
commit 1433eca29a
2 files changed
+8 -2

No files matched your search

@@ -45,6 +45,7 @@ pub struct ByQueueStats {
pub pending_face_animation_jobs: u64,
pub pending_rvc_jobs: u64,
pub pending_svc_jobs: u64,
pub pending_tacotron2_jobs: u64,
}
#[derive(Debug)]
@@ -95,6 +96,7 @@ pub async fn get_unified_queue_stats_handler(
pending_face_animation_jobs: 10_000,
pending_rvc_jobs: 10_000,
pending_svc_jobs: 10_000,
pending_tacotron2_jobs: 10_000,
}
},
legacy_tts: LegacyQueueDetails {
@@ -194,6 +196,7 @@ pub async fn get_unified_queue_stats_handler(
pending_face_animation_jobs: cacheable_stats_result.queues.sad_talker,
pending_rvc_jobs: cacheable_stats_result.queues.rvc_v2,
pending_svc_jobs: cacheable_stats_result.queues.so_vits_svc,
pending_tacotron2_jobs: cacheable_stats_result.queues.tacotron2,
}
},
legacy_tts: LegacyQueueDetails {
@@ -21,11 +21,12 @@ impl Default for CacheableQueueStats {
#[derive(Serialize, Deserialize, Clone, Default)]
pub struct Queues {
pub legacy_tts: u64, // Tacotron2
pub total_generic: u64, // Everything except Tacotron2 summed together
pub legacy_tts: u64, // Tacotron2 (as tts_inference_jobs)
pub total_generic: u64, // Everything except tts_inference_jobs' Tacotron2 summed together
pub rvc_v2: u64,
pub sad_talker: u64,
pub so_vits_svc: u64,
pub tacotron2: u64, // Tacotron2 (as generic_inference_jobs; NB: included in `total_generic`)
}
pub fn database_result_to_cacheable(database_records: Vec<QueueStatsRow>) -> CacheableQueueStats {
@@ -43,6 +44,7 @@ pub fn database_result_to_cacheable(database_records: Vec<QueueStatsRow>) -> Cac
let rvc_v2 = unwrap(queue_lengths.get("rvc_v2"));
let sad_talker = unwrap(queue_lengths.get("sad_talker"));
let so_vits_svc = unwrap(queue_lengths.get("so_vits_svc"));
let tacotron2 = unwrap(queue_lengths.get("tacotron2"));
let total_generic = rvc_v2 + so_vits_svc;
@@ -54,6 +56,7 @@ pub fn database_result_to_cacheable(database_records: Vec<QueueStatsRow>) -> Cac
rvc_v2,
sad_talker,
so_vits_svc,
tacotron2,
},
}
}