Update gen_images.py

gen_images.py

@@ -19,7 +19,7 @@ import PIL.Image
 import torch
 
 import legacy
-os.environ["CUDA_VISIBLE_DEVICES"] = ""  # Disables CUDA completely
+
 
 
 #----------------------------------------------------------------------------
@@ -105,7 +105,7 @@ def generate_images(
     """
 
     print('Loading networks from "%s"...' % network_pkl)
-    device = torch.device('cpu')
+    device = torch.device('cuda')
     with dnnlib.util.open_url(network_pkl) as f:
         G = legacy.load_network_pkl(f)['G_ema'].to(device) # type: ignore