Merge pull request #388 from christinaexyou/convert-tensor-dtype

Convert tensor dtype to float64

caikit_nlp/modules/text_classification/sequence_classification.py

@@ -185,7 +185,7 @@ def _get_scores(self, text: Union[str, List[str]]):
 
         softmax = torch.nn.Softmax(dim=1)
         raw_scores = softmax(logits)
-        scores = raw_scores.numpy()
+        scores = raw_scores.double().numpy()
         num_labels = self.model.num_labels
         num_texts = 1  # str
         if isinstance(text, List):