| { | |
| "model_type": "kinase_substrate_bert", | |
| "pipeline_tag": "text-classification", | |
| "tape_model_name": "bert-base", | |
| "num_labels": 2, | |
| "threshold": 0.7734375, | |
| "with_sep": true, | |
| "max_len": 1024, | |
| "auto_map": { | |
| "AutoConfig": "modeling.KinaseSubstrateConfig", | |
| "AutoModel": "modeling.KinaseSubstrateModel" | |
| }, | |
| "custom_pipelines": { | |
| "kinase-substrate": { | |
| "impl": "pipeline.KinaseSubstratePipeline", | |
| "pt": [ | |
| "AutoModel" | |
| ] | |
| } | |
| } | |
| } |