add option for gpu in settings

2023-08-06 00:38:30 +02:00
parent 76dbd57ad5
commit 0db1f452fd
7 changed files with 182 additions and 136 deletions
@@ -16,6 +16,16 @@ def on_ui_settings() -> None:
            section=section,
        ),
    )
+    shared.opts.add_option(
+        "faceswaplab_use_gpu",
+        shared.OptionInfo(
+            False,
+            "Use GPU, only for CUDA on Windows/Linux - experimental and risky, can messed up dependencies (requires restart)",
+            gr.Checkbox,
+            {"interactive": True},
+            section=section,
+        ),
+    )
    shared.opts.add_option(
        "faceswaplab_keep_original",
        shared.OptionInfo(