No commit message

modules/shared.py

@@ -145,15 +145,6 @@
 if devices.adl is None or devices.hMEM is None or (cmd_opts.device_id is not None and cmd_opts.device_id != "0"):
     print("Disabled experimental graphic memory optimizations.")
     cmd_opts.disable_experimental_memopt = True
-if not cmd_opts.no_half:
-    import torch
-    torch.cat = devices.cat
-
-    torch.nn.GroupNorm = devices.GroupNorm
-    torch.nn.LayerNorm = devices.LayerNorm
-    torch.nn.Linear = devices.Linear
-    torch.nn.Conv2d = devices.Conv2d
-    torch.nn.functional.pad = devices.pad
 
 
 devices.device, devices.device_interrogate, devices.device_gfpgan, devices.device_esrgan, devices.device_codeformer = \
@@ -174,6 +165,17 @@
 loaded_hypernetworks = []
 
 
+if not cmd_opts.no_half and device.type == 'privateuseone':
+    import torch
+    torch.cat = devices.cat
+
+    torch.nn.GroupNorm = devices.GroupNorm
+    torch.nn.LayerNorm = devices.LayerNorm
+    torch.nn.Linear = devices.Linear
+    torch.nn.Conv2d = devices.Conv2d
+    torch.nn.functional.pad = devices.pad
+
+
 def reload_hypernetworks():
     from modules.hypernetworks import hypernetwork
     global hypernetworks