{ | |
"architectures": [ | |
"QBModelWrapper" | |
], | |
"auto_map": { | |
"AutoConfig": "QBModelConfig.QBModelConfig", | |
"AutoModelForQuestionAnswering": "QBModelWrapperCopy.QBModelWrapper" | |
}, | |
"custom_pipelines": { | |
"qa-pipeline-qb": { | |
"impl": "QBpipeline.QApipeline", | |
"pt": [ | |
"AutoModelForQuestionAnswering" | |
], | |
"tf": [ | |
"TFAutoModelForQuestionAnswering" | |
] | |
} | |
}, | |
"model_type": "QA-umd-quizbowl", | |
"torch_dtype": "float16", | |
"transformers_version": "4.40.1" | |
} | |