| { | |
| "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" | |
| } | |