diff --git a/util/face_sdk/core/model_loader/face_detection/FaceDetModelLoader.py b/util/face_sdk/core/model_loader/face_detection/FaceDetModelLoader.py index 5df468e..e7d7a0b 100644 --- a/util/face_sdk/core/model_loader/face_detection/FaceDetModelLoader.py +++ b/util/face_sdk/core/model_loader/face_detection/FaceDetModelLoader.py @@ -28,5 +28,5 @@ def __init__(self, model_path, model_category, model_name, meta_file='model_meta self.cfg['confidence_threshold'] = self.meta_conf['confidence_threshold'] def load_model(self): - model = torch.load(self.cfg['model_file_path']) + model = torch.load(self.cfg['model_file_path'], weights_only=False) return model, self.cfg