There was an error while loading. Please reload this page.
2 parents d744148 + cfecb3e commit 4c355d8Copy full SHA for 4c355d8
1 file changed
manga_translator/inpainting/ldm/modules/midas/midas/base_model.py
@@ -8,7 +8,7 @@ def load(self, path):
8
Args:
9
path (str): file path
10
"""
11
- parameters = torch.load(path, map_location=torch.device('cpu'))
+ parameters = torch.load(path, map_location=torch.device('cpu'), weights_only=True)
12
13
if "optimizer" in parameters:
14
parameters = parameters["model"]
0 commit comments