diff --git a/deepdoc/parser/pdf_parser.py b/deepdoc/parser/pdf_parser.py index e5268a8a4..ab888eb2a 100644 --- a/deepdoc/parser/pdf_parser.py +++ b/deepdoc/parser/pdf_parser.py @@ -45,9 +45,12 @@ class RAGFlowPdfParser: self.updown_cnt_mdl = xgb.Booster() if not LIGHTEN: - import torch - if torch.cuda.is_available(): - self.updown_cnt_mdl.set_param({"device": "cuda"}) + try: + import torch + if torch.cuda.is_available(): + self.updown_cnt_mdl.set_param({"device": "cuda"}) + except Exception as e: + logging.error(str(e)) try: model_dir = os.path.join( get_project_base_directory(),