nes470
/

system-with-gen-pipeline

Text2Text Generation

QA-umd-quizbowl

Model card Files Files and versions Community

nes470 commited on May 14

Commit

9c7092d

•

1 Parent(s): e81aef3

Update QBGenModelWrapper.py

Files changed (1) hide show

QBGenModelWrapper.py +2 -2

QBGenModelWrapper.py CHANGED Viewed

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

 from typing import List
 from transformers import PreTrainedModel
 from transformers import PretrainedConfig
+from .QBModelConfig import QBModelConfig
+from .qbmodel import QuizBowlModel
 class QBGenModelWrapper(PreTrainedModel):
     config_class= QBModelConfig