reward-model / config.json
Panga-Azazia's picture
Training in progress, epoch 0
f39b358 verified
raw
history blame contribute delete
774 Bytes
{
"architectures": [
"RewardModel"
],
"dtype": "float32",
"model_type": "reward-model",
"override": {},
"pretrained_config": {
"audio_conv_channels": 128,
"audio_conv_layers": 3,
"chunk_length": 30,
"dither": 0.0,
"dropout": 0.1,
"embed_dim": 128,
"head_hidden": 256,
"hop_length": 256,
"kernel_size": 5,
"lstm_hidden": 128,
"lstm_layers": 1,
"n_fft": 1024,
"n_mel": 80,
"pad_token_id": 1,
"padding": 2,
"padding_value": 0,
"return_attention_mask": true,
"sample_rate": 16000,
"stride": 1,
"vocab_size": 2048
},
"tokenizer_path": "bambara-asr/tokenizers/bayelemabaga/tokenizer.model",
"torch_dtype": "float32",
"transformers_version": "4.53.3",
"use_pre": "default"
}