rodrigomasini commited on
Commit
623dc9f
·
verified ·
1 Parent(s): 2c94343

Update mdr_pdf_parser.py

Browse files
Files changed (1) hide show
  1. mdr_pdf_parser.py +4 -0
mdr_pdf_parser.py CHANGED
@@ -69,6 +69,10 @@ try:
69
  except ImportError:
70
  print("Warning: Could not import build_model from struct_eqtable. Table parsing might fail.")
71
 
 
 
 
 
72
  # --- MagicDataReadiness Core Components ---
73
 
74
  # --- MDR Utilities ---
 
69
  except ImportError:
70
  print("Warning: Could not import build_model from struct_eqtable. Table parsing might fail.")
71
 
72
+ import torch
73
+ if not hasattr(torch, "get_default_device"):
74
+ torch.get_default_device = lambda: torch.device("cuda" if torch.cuda.is_available() else "cpu")
75
+
76
  # --- MagicDataReadiness Core Components ---
77
 
78
  # --- MDR Utilities ---