File size: 506 Bytes
ec31fb7 e81aef3 ec31fb7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
{
"architectures": [
"QBGenModelWrapper"
],
"auto_map": {
"AutoConfig": "QBModelConfig.QBModelConfig",
"AutoModelForSeq2SeqLM": "QBGenModelWrapper.QBGenModelWrapper"
},
"custom_pipelines": {
"gen-qa-pipeline-qb": {
"impl": "QBGenPipeline.GenPipeline",
"pt": [
"AutoModelForSeq2SeqLM"
],
"tf": [
"TFAutoModelForSeq2SeqLM"
]
}
},
"model_type": "QA-umd-quizbowl",
"torch_dtype": "float16",
"transformers_version": "4.40.1"
}
|