Upload GenPipeline

QBGenModelWrapper.py

@@ -1,8 +1,8 @@
 from typing import List
 from transformers import PreTrainedModel
 from transformers import PretrainedConfig
-from .QBModelConfig import QBModelConfig
-from .qbmodel import QuizBowlModel
+from QBModelConfig import QBModelConfig
+from qbmodel import QuizBowlModel
 
 class QBGenModelWrapper(PreTrainedModel):
     config_class= QBModelConfig

README.md

@@ -1,6 +1,6 @@
 ---
-library_name: transformers
 license: mit
+library_name: transformers
 ---
 
 # Model Card for Model ID

config.json

@@ -4,7 +4,7 @@
   ],
   "auto_map": {
     "AutoConfig": "QBModelConfig.QBModelConfig",
-    "AutoModelForQuestionAnswering": "QBGenModelWrapper.QBGenModelWrapper"
+    "AutoModelForSeq2SeqLM": "QBGenModelWrapper.QBGenModelWrapper"
   },
   "custom_pipelines": {
     "gen-qa-pipeline-qb": {