Reduce gpu_mem_limit to 1GB

adaface/face_id_to_ada_prompt.py

@@ -661,7 +661,7 @@ class Arc2Face_ID2AdaPrompt(FaceID2AdaPrompt):
                                          providers=['CUDAExecutionProvider'],
                                          provider_options=[{"device_id": device_id, 
                                                             "cudnn_conv_algo_search": "HEURISTIC",
-                                                            "gpu_mem_limit": 2 * 1024**3}])
+                                                            "gpu_mem_limit": 1 * 1024**3}])
             self.face_app.prepare(ctx_id=device_id, det_size=(256, 256))
 
         self.device = device
@@ -808,7 +808,7 @@ class ConsistentID_ID2AdaPrompt(FaceID2AdaPrompt):
                                          providers=['CUDAExecutionProvider'],
                                          provider_options=[{"device_id": device_id, 
                                                             "cudnn_conv_algo_search": "HEURISTIC",
-                                                            "gpu_mem_limit": 2 * 1024**3}])
+                                                            "gpu_mem_limit": 1 * 1024**3}])
             self.face_app.prepare(ctx_id=device_id, det_size=(256, 256))
 
         self.device = device