Update difpoint/src/models/predictor.py

difpoint/src/models/predictor.py

@@ -227,7 +227,10 @@ class OnnxRuntimePredictor:
             if self.inputs[i].type == 'tensor(float16)':
                 input_feeds[self.inputs[i].name] = data[i].astype(np.float16)
             else:
-                input_feeds[self.inputs[i].name] = data[i].astype(np.float32)
+                try:
+                    input_feeds[self.inputs[i].name] = data[i].astype(np.float32)
+                except:
+                    input_feeds[self.inputs[i].name] = data[i].cpu().numpy().astype(np.float32)
         results = self.onnx_model.run(None, input_feeds)
         return results