Hugging Face's logo

Backedman
/
TriviaAnsweringMachine4

Question Answering
Transformers
PyTorch
Safetensors
TFIDF-QA
custom_code
Model card Files
xet
Community
Backedman commited on
Commit
8791afc
路
verified 路
1 Parent(s): 754a47d

Upload model

Browse files
Files changed (2) hide show
  1. QBModelConfig.py +1 -1
  2. config.json +1 -1
QBModelConfig.py CHANGED
@@ -6,4 +6,4 @@ class QBModelConfig(PretrainedConfig):
6
 
7
  def __init__(self, **kwargs):
8
  super().__init__( **kwargs)
9
- self.torch_dtype = "torch.float16"
 
6
 
7
  def __init__(self, **kwargs):
8
  super().__init__( **kwargs)
9
+ self.torch_dtype = torch.float16
config.json CHANGED
@@ -7,6 +7,6 @@
7
  "AutoModelForQuestionAnswering": "QBModelWrapper.QBModelWrapper"
8
  },
9
  "model_type": "TFIDF-QA",
10
- "torch_dtype": "torch.float16",
11
  "transformers_version": "4.40.1"
12
  }
 
7
  "AutoModelForQuestionAnswering": "QBModelWrapper.QBModelWrapper"
8
  },
9
  "model_type": "TFIDF-QA",
10
+ "torch_dtype": "float16",
11
  "transformers_version": "4.40.1"
12
  }