Skip to content

Commit af65796

Browse files
committed
fix: fixed issue #527
1 parent 66d5a04 commit af65796

File tree

1 file changed

+1
-1
lines changed
  • python/rapidocr/inference_engine

1 file changed

+1
-1
lines changed

python/rapidocr/inference_engine/torch.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ def __init__(self, cfg) -> None:
5353

5454
arch_config = all_arch_config.get(file_name)
5555
self.predictor = BaseModel(arch_config)
56-
self.predictor.load_state_dict(torch.load(model_path, weights_only=True))
56+
self.predictor.load_state_dict(torch.load(model_path, weights_only=False))
5757
self.predictor.eval()
5858

5959
self.use_gpu = False

0 commit comments

Comments
 (0)