We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7478723 commit b7c5e60Copy full SHA for b7c5e60
modules/dml/hijack/torch.py
@@ -52,7 +52,7 @@ def pow_(self: torch.Tensor, *args, **kwargs):
52
53
54
_load = torch.load
55
-def load(f, map_location, *args, **kwargs):
+def load(f, map_location = "cpu", *args, **kwargs):
56
if type(map_location) in (str, torch.device,):
57
device = torch.device(map_location)
58
if device.type == "privateuseone":
0 commit comments