aha 0.2.6

aha model inference library, now supports Qwen(2.5VL/3/3VL/3.5/ASR/3Embedding/3Reranker), MiniCPM(4/5), VoxCPM(0.5B/1.5/2), DeepSeek-OCR/2, Hunyuan-OCR, PaddleOCR-VL/1.5, RMBG2.0, GLM(ASR-Nano-2512/OCR), Fun-ASR-Nano-2512, LFM(2/2.5/2VL/2.5VL)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

use serde::Deserialize;

#[derive(Debug, Deserialize)]
pub struct MossAudioTokenizerConfig {
    pub sample_rate: usize,
    pub sampling_rate: usize,
    pub downsample_rate: usize,
    pub causal_transformer_context_duration: f64,
    pub number_channels: usize,
    pub enable_channel_interleave: bool,
    pub compute_dtype: String,
    pub dtype: String,
    pub code_dim: usize,
    pub encoder_kwargs: Vec<MossAudioTokenizerModuleConfig>,
    pub decoder_kwargs: Vec<MossAudioTokenizerModuleConfig>,
    pub quantizer_type: String,
    pub quantizer_kwargs: MossAudioTokenizerQuantizerKwargs,
    pub reversed_decoder_kwargs: Vec<MossAudioTokenizerModuleConfig>,
}

#[derive(Debug, Deserialize)]
pub struct MossAudioTokenizerModuleConfig {
    pub module_type: String,
    pub patch_size: Option<usize>,
    pub causal: Option<bool>,
    pub context_duration: Option<f64>,
    pub conv_layout: Option<bool>,
    pub d_model: Option<usize>,
    pub dim_feedforward: Option<usize>,
    pub gating: Option<String>,
    pub input_dimension: Option<usize>,
    pub layer_scale: Option<f64>,
    pub max_period: Option<usize>,
    pub norm: Option<String>,
    pub num_heads: Option<usize>,
    pub num_layers: Option<usize>,
    pub output_dimension: Option<usize>,
    pub positional_embedding: Option<String>,
}

#[derive(Debug, Deserialize)]
pub struct MossAudioTokenizerQuantizerKwargs {
    pub codebook_dim: usize,
    pub codebook_loss_weight: f64,
    pub codebook_size: usize,
    pub commitment_loss_weight: f64,
    pub input_dim: usize,
    pub num_quantizers: usize,
    pub output_dim: usize,
    pub quantizer_dropout: f64,
    pub quantizer_type: String,
    pub rvq_dim: usize,
}