File size: 2,787 Bytes
37c33c8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
{
    "architectures": [
        "GraniteSpeechForConditionalGeneration"
    ],
    "audio_token_index": 49155,
    "auto_map": {
        "AutoConfig": "configuration_granite_speech.GraniteSpeechConfig",
        "AutoFeatureExtractor": "feature_extraction_granite_speech.GraniteSpeechFeatureExtractor",
        "AutoModelForSpeechSeq2Seq": "modeling_granite_speech.GraniteSpeechForConditionalGeneration",
        "AutoProcessor": "processing_granite_speech.GraniteSpeechProcessor"
    },
    "encoder_config": {
        "context_size": 200,
        "conv_expansion_factor": 2,
        "conv_kernel_size": 15,
        "dim_head": 128,
        "dropout": 0.1,
        "feedforward_mult": 4,
        "hidden_dim": 1024,
        "input_dim": 160,
        "model_type": "granite_speech_encoder",
        "num_heads": 8,
        "num_layers": 10,
        "output_dim": 42
    },
    "has_lora_adapter": true,
    "initializer_range": 0.02,
    "model_type": "granite_speech",
    "projector_config": {
        "_attn_implementation_autoset": true,
        "attention_probs_dropout_prob": 0.1,
        "cross_attention_frequency": 1,
        "downsample_rate": 5,
        "encoder_hidden_size": 1024,
        "hidden_act": "gelu",
        "hidden_dropout_prob": 0.1,
        "hidden_size": 1024,
        "initializer_range": 0.02,
        "intermediate_size": 4096,
        "layer_norm_eps": 1e-12,
        "llm_dim": 4096,
        "max_position_embeddings": 2048,
        "model_type": "granite_speech_qformer",
        "num_attention_heads": 16,
        "num_hidden_layers": 2,
        "position_embedding_type": "absolute",
        "use_qformer_text_input": false,
        "window_size": 15
    },
    "text_config": {
        "_name_or_path": "ibm-granite/granite-3.2-8b-instruct",
        "architectures": [
            "GraniteForCausalLM"
        ],
        "attention_bias": false,
        "attention_dropout": 0.0,
        "attention_multiplier": 0.0078125,
        "bos_token_id": 0,
        "embedding_multiplier": 12.0,
        "eos_token_id": 0,
        "hidden_act": "silu",
        "hidden_size": 4096,
        "initializer_range": 0.02,
        "intermediate_size": 12800,
        "logits_scaling": 16.0,
        "max_position_embeddings": 131072,
        "mlp_bias": false,
        "model_type": "granite",
        "num_attention_heads": 32,
        "num_hidden_layers": 40,
        "num_key_value_heads": 8,
        "pad_token_id": 0,
        "residual_multiplier": 0.22,
        "rms_norm_eps": 1e-05,
        "rope_scaling": null,
        "rope_theta": 10000000.0,
        "tie_word_embeddings": true,
        "torch_dtype": "bfloat16",
        "use_cache": true,
        "vocab_size": 49156
    },
    "torch_dtype": "bfloat16",
    "transformers_version": "4.50.0.dev0"
}