Fix lshqqytiger#449.

IllusiveMan196 · Apr 19, 2024 · b7c5e60 · b7c5e60
1 parent 7478723
commit b7c5e60
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/dml/hijack/torch.py b/modules/dml/hijack/torch.py
@@ -52,7 +52,7 @@ def pow_(self: torch.Tensor, *args, **kwargs):
 
 
 _load = torch.load
-def load(f, map_location, *args, **kwargs):
+def load(f, map_location = "cpu", *args, **kwargs):
     if type(map_location) in (str, torch.device,):
         device = torch.device(map_location)
         if device.type == "privateuseone":