From 65fbefd0337f9deb913c6956a9cfe2155c9c2f5b Mon Sep 17 00:00:00 2001
From: xeonvs <xeonvs@gmail.com>
Date: Wed, 7 Sep 2022 15:58:25 +0200
Subject: Added support for launching on Apple Silicon

---
 modules/esrgan_model.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'modules/esrgan_model.py')

diff --git a/modules/esrgan_model.py b/modules/esrgan_model.py
index 3dcef5a6..2ed1d273 100644
--- a/modules/esrgan_model.py
+++ b/modules/esrgan_model.py
@@ -14,8 +14,11 @@ import modules.images
 
 def load_model(filename):
     # this code is adapted from https://github.com/xinntao/ESRGAN
-
-    pretrained_net = torch.load(filename)
+    if torch.has_mps:
+        map_l = 'cpu'
+    else:
+        map_l = None
+    pretrained_net = torch.load(filename, map_location=map_l)
     crt_model = arch.RRDBNet(3, 3, 64, 23, gc=32)
 
     if 'conv_first.weight' in pretrained_net:
-- 
cgit v1.2.1