Skip to content

Commit b7c5e60

Browse files
committed
1 parent 7478723 commit b7c5e60

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

modules/dml/hijack/torch.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def pow_(self: torch.Tensor, *args, **kwargs):
5252

5353

5454
_load = torch.load
55-
def load(f, map_location, *args, **kwargs):
55+
def load(f, map_location = "cpu", *args, **kwargs):
5656
if type(map_location) in (str, torch.device,):
5757
device = torch.device(map_location)
5858
if device.type == "privateuseone":

0 commit comments

Comments
 (0)