How to use maxwlnd/cl_mention_embedding with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="maxwlnd/cl_mention_embedding")
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("maxwlnd/cl_mention_embedding", dtype="auto")
fd5f978 9263158 fd5f978
1
2
3
4
5
6
7
8
{ "model_type": "bert", "pretrained_model_name": "bert-base-uncased", "hidden_size": 768, "proj_dim": 128, "prompt_template": "Social group of {} is: [MASK].", "max_length": 128 }
