change interface

interface.py

@@ -19,25 +19,21 @@ with open(cached_download(hf_hub_url('ykilcher/apes', 'gamma500/network-snapshot
     G = pickle.load(f)["G_ema"]  # torch.nn.Module
 
 device = torch.device("cpu")
-if false:
-    device = torch.device("cuda")
-    G = G.to(device)
-else:
-    _old_forward = G.forward
+_old_forward = G.forward
 
-    def _new_forward(self, *args, **kwargs):
-        kwargs["force_fp32"] = True
-        return _old_forward(*args, **kwargs)
+def _new_forward(self, *args, **kwargs):
+    kwargs["force_fp32"] = True
+    return _old_forward(*args, **kwargs)
 
-    G.forward = types.MethodType(_new_forward, G)
+G.forward = types.MethodType(_new_forward, G)
 
-    _old_synthesis_forward = G.synthesis.forward
+_old_synthesis_forward = G.synthesis.forward
 
-    def _new_synthesis_forward(self, *args, **kwargs):
-        kwargs["force_fp32"] = True
-        return _old_synthesis_forward(*args, **kwargs)
+def _new_synthesis_forward(self, *args, **kwargs):
+    kwargs["force_fp32"] = True
+    return _old_synthesis_forward(*args, **kwargs)
 
-    G.synthesis.forward = types.MethodType(_new_synthesis_forward, G.synthesis)
+G.synthesis.forward = types.MethodType(_new_synthesis_forward, G.synthesis)
 
 
 def generate(