| { | |
| "model_type": "distilbert", | |
| "num_labels": 2, | |
| "pipeline_tag": "text-classification", | |
| "vocab_size": 30522, | |
| "max_position_embeddings": 512, | |
| "dim": 64, | |
| "hidden_dim": 256, | |
| "n_layers": 2, | |
| "n_heads": 2, | |
| "dropout": 0.1, | |
| "custom_pipelines": { | |
| "admesh-intent": { | |
| "impl": "pipeline.AdmeshIntentPipeline", | |
| "pt": [ | |
| "AutoModelForSequenceClassification" | |
| ], | |
| "tf": [] | |
| } | |
| } | |
| } | |