Upload GenPipeline
Browse files- QBGenModelWrapper.py +2 -2
- README.md +1 -1
- config.json +1 -1
QBGenModelWrapper.py
CHANGED
@@ -1,8 +1,8 @@
|
|
1 |
from typing import List
|
2 |
from transformers import PreTrainedModel
|
3 |
from transformers import PretrainedConfig
|
4 |
-
from
|
5 |
-
from
|
6 |
|
7 |
class QBGenModelWrapper(PreTrainedModel):
|
8 |
config_class= QBModelConfig
|
|
|
1 |
from typing import List
|
2 |
from transformers import PreTrainedModel
|
3 |
from transformers import PretrainedConfig
|
4 |
+
from QBModelConfig import QBModelConfig
|
5 |
+
from qbmodel import QuizBowlModel
|
6 |
|
7 |
class QBGenModelWrapper(PreTrainedModel):
|
8 |
config_class= QBModelConfig
|
README.md
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
---
|
2 |
-
library_name: transformers
|
3 |
license: mit
|
|
|
4 |
---
|
5 |
|
6 |
# Model Card for Model ID
|
|
|
1 |
---
|
|
|
2 |
license: mit
|
3 |
+
library_name: transformers
|
4 |
---
|
5 |
|
6 |
# Model Card for Model ID
|
config.json
CHANGED
@@ -4,7 +4,7 @@
|
|
4 |
],
|
5 |
"auto_map": {
|
6 |
"AutoConfig": "QBModelConfig.QBModelConfig",
|
7 |
-
"
|
8 |
},
|
9 |
"custom_pipelines": {
|
10 |
"gen-qa-pipeline-qb": {
|
|
|
4 |
],
|
5 |
"auto_map": {
|
6 |
"AutoConfig": "QBModelConfig.QBModelConfig",
|
7 |
+
"AutoModelForSeq2SeqLM": "QBGenModelWrapper.QBGenModelWrapper"
|
8 |
},
|
9 |
"custom_pipelines": {
|
10 |
"gen-qa-pipeline-qb": {
|