| { | |
| "add_ffn_bias": false, | |
| "add_lm_head_bias": true, | |
| "add_qkv_bias": true, | |
| "architectures": [ | |
| "OpenBAForConditionalGeneration" | |
| ], | |
| "auto_map": { | |
| "AutoConfig": "configuration_openba.OpenBAConfig", | |
| "AutoModel": "modeling_openba.OpenBAForConditionalGeneration", | |
| "AutoModelForCausalLM": "modeling_openba.OpenBAForConditionalGeneration", | |
| "AutoModelForSeq2SeqLM": "modeling_openba.OpenBAForConditionalGeneration" | |
| }, | |
| "attention_dropout": 0.1, | |
| "decoder_max_seq_length": 256, | |
| "decoder_start_token_id": 0, | |
| "eos_token_id": 1, | |
| "ffn_hidden_size": 16384, | |
| "hidden_dropout": 0.1, | |
| "hidden_size": 4096, | |
| "initializer_factor": 1.0, | |
| "is_encoder_decoder": true, | |
| "kv_channels": 128, | |
| "max_seq_length": 1024, | |
| "model_type": "openba", | |
| "num_decoder_layers": 36, | |
| "num_heads": 40, | |
| "num_layers": 12, | |
| "pad_token_id": 0, | |
| "tie_word_embeddings": false, | |
| "tokenizer_class": "OpenBATokenizer", | |
| "transformers_version": "4.31.0", | |
| "use_cache": true, | |
| "vocab_size": 250368 | |
| } | |