Update roop/core.py

roop/core.py

@@ -67,7 +67,7 @@ def parse_args() -> None:
     globals.temp_frame_quality = args.temp_frame_quality
     globals.output_video_encoder = args.output_video_encoder
     globals.output_video_quality = args.output_video_quality
-    globals.max_memory = 4 #args.max_memory
+    globals.max_memory = args.max_memory
     globals.execution_providers = decode_execution_providers(args.execution_provider)
     globals.execution_threads = args.execution_threads
 
@@ -96,7 +96,7 @@ def limit_resources() -> None:
     gpus = tensorflow.config.experimental.list_physical_devices('GPU')
     for gpu in gpus:
         tensorflow.config.experimental.set_virtual_device_configuration(gpu, [
-            tensorflow.config.experimental.VirtualDeviceConfiguration(memory_limit=1024)
+            tensorflow.config.experimental.VirtualDeviceConfiguration(memory_limit=2048)
         ])
     # limit memory usage
     if globals.max_memory: