mentee-embed-v3 / config.json
SyedSyab's picture
Add HF AutoModel/AutoTokenizer support (trust_remote_code=True)
f4f4a61 verified
Raw
History Blame Contribute Delete
532 Bytes
{
"activation": "gelu",
"architectures": [
"MenteeEmbedModel"
],
"auto_map": {
"AutoConfig": "configuration_mentee.MenteeEmbedConfig",
"AutoModel": "modeling_mentee.MenteeEmbedModel"
},
"dropout": 0.1,
"ffn": 1536,
"heads": 12,
"hidden": 384,
"hidden_size": 384,
"layers": 12,
"max_position": 512,
"max_position_embeddings": 512,
"model_type": "mentee_embed",
"pooling": "mean",
"torch_dtype": "float32",
"transformers_version": "4.46.3",
"vocab_size": 50000
}