musicgpt 0.3.1

Generate music samples from natural language prompt locally with your own computer
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

use serde::{Deserialize, Serialize};

#[derive(Serialize, Deserialize)]
pub struct MusicGenConfig {
    pub audio_encoder: AudioEncoderConfig,
    pub decoder: DecoderConfig,
    pub text_encoder: TextEncoderConfig,
}

#[derive(Serialize, Deserialize)]
pub struct AudioEncoderConfig {
    pub sampling_rate: usize,
}

#[derive(Serialize, Deserialize)]
pub struct DecoderConfig {
    pub num_attention_heads: usize,
    pub num_hidden_layers: usize,
    pub top_k: usize,
    pub pad_token_id: i64,
}

#[derive(Serialize, Deserialize)]
pub struct TextEncoderConfig {
    pub d_kv: usize,
}