| { | |
| "architectures": [ | |
| "SimilarityModel" | |
| ], | |
| "auto_map": { | |
| "AutoConfig": "configuration.SimilarityModelConfig", | |
| "AutoModel": "similarity_model.SimilarityModel" | |
| }, | |
| "device": "cpu", | |
| "embedding_model_config": { | |
| "add_upper": true, | |
| "d": 128, | |
| "device": "cpu", | |
| "model_class": "roberta", | |
| "model_name": "roberta-base", | |
| "normalize": true, | |
| "pooling": "pooler", | |
| "prompt": "", | |
| "upper_case": false | |
| }, | |
| "model_type": "roberta", | |
| "score_model_config": { | |
| "alpha": 50 | |
| }, | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.27.4", | |
| "weighting_function_config": { | |
| "weighting_exponent": 0.5 | |
| } | |
| } | |