Comet-1-Preview / config.json
Bc-AI's picture
Add Transformers support (modeling + config code), tokenizer/generation configs and model card
9df61c9 verified
Raw History Blame Contribute Delete
982 Bytes
{
"architectures": [
"CometForCausalLM"
],
"auto_map": {
"AutoConfig": "configuration_comet.CometConfig",
"AutoModel": "modeling_comet.CometModel",
"AutoModelForCausalLM": "modeling_comet.CometForCausalLM"
},
"model_type": "comet",
"variant": "7.0b",
"vocab_size": 50277,
"hidden_size": 4096,
"num_hidden_layers": 36,
"num_attention_heads": 32,
"num_key_value_heads": 8,
"intermediate_size": 12288,
"max_position_embeddings": 8192,
"rope_theta": 1000000.0,
"memory_width": 2048,
"memory_kernel_size": 7,
"memory_dilations": [1, 2, 4, 8],
"layer_pattern": ["global", "global", "global", "recall"],
"layer_scale_init": 0.5,
"qk_norm": true,
"rms_norm_eps": 1e-06,
"fused_projections": true,
"dropout": 0.0,
"initializer_range": 0.02,
"tie_word_embeddings": true,
"use_cache": true,
"bos_token_id": 0,
"eos_token_id": 0,
"pad_token_id": 1,
"tokenizer_class": "PreTrainedTokenizerFast",
"dtype": "float32"
}