zero-shot-embedding / config.json
iamroot's picture
Upload ZeroShotEmbedding
def591b
raw
history blame
379 Bytes
{
"architectures": [
"ZeroShotEmbedding"
],
"auto_map": {
"AutoConfig": "model.ZeroShotEmbeddingConfig",
"AutoModel": "model.ZeroShotEmbedding"
},
"base_embedding_model": "all-mpnet-base-v2",
"hidden_size": 2048,
"input_size": 768,
"model_type": "embedding-head",
"output_size": 128,
"torch_dtype": "float32",
"transformers_version": "4.35.0"
}