Update run_inference.py

run_inference.py

@@ -4,7 +4,7 @@ import torch
 
 # Allow unpickling of HistoryBuffer
 from mmengine.logging.history_buffer import HistoryBuffer
-# Allow numpy reconstruct, ndarray, dtype, and Float64DType
+# Allow numpy reconstruct, ndarray, dtype, Float64DType, Int64DType, and getattr
 import numpy as np
 import numpy.core.multiarray as cam
 import numpy.dtypes as dtypes
@@ -16,6 +16,7 @@ torch.serialization.add_safe_globals([
     np.dtype,
     dtypes.Float64DType,
     dtypes.Int64DType,
+    getattr,
 ])
 
 import torch.nn as nn