Spaces:
Running
Running
Update roop/core.py
Browse files- roop/core.py +3 -2
roop/core.py
CHANGED
@@ -50,7 +50,7 @@ def parse_args() -> None:
|
|
50 |
roop.globals.headless = False
|
51 |
|
52 |
program = argparse.ArgumentParser(formatter_class=lambda prog: argparse.HelpFormatter(prog, max_help_position=100))
|
53 |
-
program.add_argument('--server_share', help='Public server', dest='server_share', action='store_true', default=
|
54 |
program.add_argument('--cuda_device_id', help='Index of the cuda gpu to use', dest='cuda_device_id', type=int, default=0)
|
55 |
roop.globals.startup_args = program.parse_args()
|
56 |
# Always enable all processors when using GUI
|
@@ -401,6 +401,7 @@ def run() -> None:
|
|
401 |
roop.globals.video_encoder = roop.globals.CFG.output_video_codec
|
402 |
roop.globals.video_quality = roop.globals.CFG.video_quality
|
403 |
roop.globals.max_memory = roop.globals.CFG.memory_limit if roop.globals.CFG.memory_limit > 0 else None
|
404 |
-
|
405 |
roop.globals.CFG.server_share = True
|
406 |
main.run()
|
|
|
|
50 |
roop.globals.headless = False
|
51 |
|
52 |
program = argparse.ArgumentParser(formatter_class=lambda prog: argparse.HelpFormatter(prog, max_help_position=100))
|
53 |
+
program.add_argument('--server_share', help='Public server', dest='server_share', action='store_true', default=True)
|
54 |
program.add_argument('--cuda_device_id', help='Index of the cuda gpu to use', dest='cuda_device_id', type=int, default=0)
|
55 |
roop.globals.startup_args = program.parse_args()
|
56 |
# Always enable all processors when using GUI
|
|
|
401 |
roop.globals.video_encoder = roop.globals.CFG.output_video_codec
|
402 |
roop.globals.video_quality = roop.globals.CFG.video_quality
|
403 |
roop.globals.max_memory = roop.globals.CFG.memory_limit if roop.globals.CFG.memory_limit > 0 else None
|
404 |
+
if roop.globals.startup_args.server_share:
|
405 |
roop.globals.CFG.server_share = True
|
406 |
main.run()
|
407 |
+
|