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

Inferenece Model Type Added.

Browse Source
This commit is contained in:
root committed 2023-09-27 12:20:08 -04:00
1 parent 2173fbc89a
commit 9efc5b8dec
1 file changed
+3
Download Patch File Download Diff File Expand all files Collapse all files
crates/_types/enums/src/by_table/generic_inference_jobs/inference_model_type.rs
+3
View File
@@ -101,6 +101,7 @@ use crate::by_table::generic_inference_jobs::inference_model_type::InferenceMode
assert_eq!(InferenceModelType::SoVitsSvc.to_str(), "so_vits_svc");
assert_eq!(InferenceModelType::Tacotron2.to_str(), "tacotron2");
assert_eq!(InferenceModelType::Vits.to_str(), "vits");
assert_eq!(InferenceModelType::VallEX.to_str(), "vall_E_X");
}
#[test]
@@ -110,6 +111,7 @@ use crate::by_table::generic_inference_jobs::inference_model_type::InferenceMode
assert_eq!(InferenceModelType::from_str("so_vits_svc").unwrap(), InferenceModelType::SoVitsSvc);
assert_eq!(InferenceModelType::from_str("tacotron2").unwrap(), InferenceModelType::Tacotron2);
assert_eq!(InferenceModelType::from_str("vits").unwrap(), InferenceModelType::Vits);
assert_eq!(InferenceModelType::from_str("vall_E_X").unwrap(), InferenceModelType::VallEX);
}
#[test]
@@ -122,6 +124,7 @@ use crate::by_table::generic_inference_jobs::inference_model_type::InferenceMode
assert_eq!(variants.pop_first(), Some(InferenceModelType::SoVitsSvc));
assert_eq!(variants.pop_first(), Some(InferenceModelType::Tacotron2));
assert_eq!(variants.pop_first(), Some(InferenceModelType::Vits));
assert_eq!(variants.pop_first(), Some(InferenceModelType::VallEX));
assert_eq!(variants.pop_first(), None);
// Generated check