Update for loading MetaTensor

Signed-off-by: Eric Kerfoot <[email protected]>
Project-MONAI · Jan 29, 2025 · 7c5bb69 · 7c5bb69
1 parent d6152a3
commit 7c5bb69
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/monai/data/meta_tensor.py b/monai/data/meta_tensor.py
@@ -609,4 +609,5 @@ def print_verbose(self) -> None:
             print(self.meta.__repr__())
 
 
-torch.serialization.add_safe_globals([MetaTensor])
+if hasattr(torch.serialization,"add_safe_globals"):
+    torch.serialization.add_safe_globals([MetaTensor])