Skip to content

Commit a7bbce5

Browse files
Update model loading to include weights_only parameter
1 parent c9d5698 commit a7bbce5

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

util/face_sdk/core/model_loader/face_detection/FaceDetModelLoader.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,5 +28,5 @@ def __init__(self, model_path, model_category, model_name, meta_file='model_meta
2828
self.cfg['confidence_threshold'] = self.meta_conf['confidence_threshold']
2929

3030
def load_model(self):
31-
model = torch.load(self.cfg['model_file_path'])
31+
model = torch.load(self.cfg['model_file_path'], weights_only=False)
3232
return model, self.cfg

0 commit comments

Comments
 (0)