param-bharat's picture
Upload DocumentSentenceRelevancePipeline
cc1fe5d verified
raw
history blame contribute delete
693 Bytes
{
"architectures": [
"MultiHeadModel"
],
"auto_map": {
"AutoConfig": "configuration.MultiHeadConfig",
"AutoModel": "modeling.MultiHeadModel"
},
"classifier_dropout": 0.1,
"custom_pipelines": {
"context-relevance": {
"impl": "pipeline.DocumentSentenceRelevancePipeline",
"pt": [
"AutoModel"
],
"tf": []
}
},
"encoder_name": "tasksource/deberta-small-long-nli",
"id2label": {
"0": "irrelevant",
"1": "relevant"
},
"label2id": {
"irrelevant": 0,
"relevant": 1
},
"model_type": "multihead",
"tokenizer_class": "DebertaV2TokenizerFast",
"torch_dtype": "float32",
"transformers_version": "4.47.0"
}