GaloNMT-7K / config.json
dupitAI's picture
Upload 6 files
4be430a verified
{
"architectures": [
"GaloNMTForConditionalGeneration"
],
"bos_token_id": 1,
"d_ff": 512,
"d_model": 256,
"decoder_start_token_id": 1,
"dropout": 0.1,
"eos_token_id": 2,
"is_encoder_decoder": true,
"max_length": 64,
"model_type": "galonmt",
"n_heads": 8,
"n_layers": 3,
"pad_token_id": 0,
"src_vocab_size": 5000,
"torch_dtype": "float32",
"transformers_version": "4.44.0",
"trg_vocab_size": 5000
}