Update roop/globals.py

roop/globals.py

@@ -1,5 +1,13 @@
 from typing import List
 
+import torch  # Import PyTorch to check for GPU availability
+
+# Check if GPU is available and set the device accordingly
+if torch.cuda.is_available():
+    device = torch.device("cuda")
+else:
+    device = torch.device("cpu")
+
 source_path = None
 target_path = None
 output_path = None