summaryrefslogtreecommitdiff
path: root/parakeet-rs/src/config.rs
diff options
context:
space:
mode:
authorsoryu <soryu@soryu.co>2025-12-21 01:27:02 +0000
committersoryu <soryu@soryu.co>2025-12-23 14:47:18 +0000
commit3c696cfc9005e73be5ed46f8941dfc8f0aca7102 (patch)
tree497bffd67001501a003739cfe0bb790502ffd50a /parakeet-rs/src/config.rs
parent55cacf6e1a087c0fa6950a1ddeb09060f787e541 (diff)
downloadsoryu-3c696cfc9005e73be5ed46f8941dfc8f0aca7102.tar.gz
soryu-3c696cfc9005e73be5ed46f8941dfc8f0aca7102.zip
Create container image and move parakeet fork to vendor dir
Diffstat (limited to 'parakeet-rs/src/config.rs')
-rw-r--r--parakeet-rs/src/config.rs51
1 files changed, 0 insertions, 51 deletions
diff --git a/parakeet-rs/src/config.rs b/parakeet-rs/src/config.rs
deleted file mode 100644
index 1dae890..0000000
--- a/parakeet-rs/src/config.rs
+++ /dev/null
@@ -1,51 +0,0 @@
-use serde::{Deserialize, Serialize};
-
-#[derive(Debug, Clone, Serialize, Deserialize)]
-pub struct PreprocessorConfig {
- pub feature_extractor_type: String,
- pub feature_size: usize,
- pub hop_length: usize,
- pub n_fft: usize,
- pub padding_side: String,
- pub padding_value: f32,
- pub preemphasis: f32,
- pub processor_class: String,
- pub return_attention_mask: bool,
- pub sampling_rate: usize,
- pub win_length: usize,
-}
-
-#[derive(Debug, Clone, Serialize, Deserialize)]
-pub struct ModelConfig {
- pub architectures: Vec<String>,
- pub vocab_size: usize,
- pub pad_token_id: usize,
-}
-
-impl Default for PreprocessorConfig {
- fn default() -> Self {
- Self {
- feature_extractor_type: "ParakeetFeatureExtractor".to_string(),
- feature_size: 80,
- hop_length: 160,
- n_fft: 512,
- padding_side: "right".to_string(),
- padding_value: 0.0,
- preemphasis: 0.97,
- processor_class: "ParakeetProcessor".to_string(),
- return_attention_mask: true,
- sampling_rate: 16000,
- win_length: 400,
- }
- }
-}
-
-impl Default for ModelConfig {
- fn default() -> Self {
- Self {
- architectures: vec!["ParakeetForCTC".to_string()],
- vocab_size: 1025,
- pad_token_id: 1024,
- }
- }
-}