[Fix]: Add TemporaryDeviceChanger for text_detection/predictor

Author: XingweiDeng

paddlex/inference/models/text_detection/predictor.py

@@ -21,6 +21,7 @@
 from ....utils.func_register import FuncRegister
 from ...common.batch_sampler import ImageBatchSampler
 from ...common.reader import ReadImage
+from ...utils.misc import is_bfloat16_available, is_float16_available
 from ..base import BasePredictor
 from ..common import ToBatch, ToCHWImage
 from .processors import DBPostProcess, DetResizeForTest, NormalizeImage
@@ -55,6 +56,15 @@ def __init__(
         self.unclip_ratio = unclip_ratio
         self.input_shape = input_shape
         self.max_side_limit = max_side_limit
+
+        self.device = kwargs.get("device", None)
+        if is_bfloat16_available(self.device):
+            self.dtype = "bfloat16"
+        elif is_float16_available(self.device):
+            self.dtype = "float16"
+        else:
+            self.dtype = "float32"
+
         self.pre_tfs, self.infer, self.post_op = self._build()
 
     def _build_batch_sampler(self):