Zhang Zelun 6 月之前
父节点
当前提交
cace3f7dc0
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      paddlex/inference/models/doc_vlm/predictor.py

+ 1 - 1
paddlex/inference/models/doc_vlm/predictor.py

@@ -47,7 +47,7 @@ class DocVLMPredictor(BasePredictor):
         self.dtype = (
         self.dtype = (
             "bfloat16"
             "bfloat16"
             if ("npu" in get_device_type() or paddle.amp.is_bfloat16_supported())
             if ("npu" in get_device_type() or paddle.amp.is_bfloat16_supported())
-            and (self.device is not None and "cpu" not in self.device)
+            and (self.device is None or "cpu" not in self.device)
             else "float32"
             else "float32"
         )
         )