Backedman
/

TriviaAnsweringMachine4

Question Answering

Model card Files Files and versions Community

Backedman commited on May 6

Commit

8791afc

•

1 Parent(s): 754a47d

Upload model

Files changed (2) hide show

QBModelConfig.py +1 -1
config.json +1 -1

QBModelConfig.py CHANGED Viewed

@@ -6,4 +6,4 @@ class QBModelConfig(PretrainedConfig):
     def __init__(self, **kwargs):
         super().__init__( **kwargs)
-        self.torch_dtype = "torch.float16"

     def __init__(self, **kwargs):
         super().__init__( **kwargs)
+        self.torch_dtype = torch.float16

config.json CHANGED Viewed

@@ -7,6 +7,6 @@
     "AutoModelForQuestionAnswering": "QBModelWrapper.QBModelWrapper"
   },
   "model_type": "TFIDF-QA",
-  "torch_dtype": "torch.float16",
   "transformers_version": "4.40.1"
 }

     "AutoModelForQuestionAnswering": "QBModelWrapper.QBModelWrapper"
   },
   "model_type": "TFIDF-QA",
+  "torch_dtype": "float16",
   "transformers_version": "4.40.1"
 }