config.json

{
  "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.1"
}