diff --git a/magic_pdf/model/pdf_extract_kit.py b/magic_pdf/model/pdf_extract_kit.py
index f4e6f9cd..46d5c3b0 100644
--- a/magic_pdf/model/pdf_extract_kit.py
+++ b/magic_pdf/model/pdf_extract_kit.py
@@ -114,6 +114,9 @@ def __init__(self, ocr: bool = False, show_log: bool = False, **kwargs):
         assert self.apply_layout, "DocAnalysis must contain layout model."
         # 初始化解析方案
         self.device = kwargs.get("device", self.configs["config"]["device"])
+        if (self.device == "direct_ml"):
+            import torch_directml
+            self.device = "privateuseone"
         logger.info("using device: {}".format(self.device))
         models_dir = kwargs.get("models_dir", os.path.join(root_dir, "resources", "models"))
         logger.info("using models_dir: {}".format(models_dir))