Try the fix on cudnn_conv_algo_search

ConsistentID/lib/pipeline_ConsistentID.py

@@ -73,9 +73,10 @@ class ConsistentIDPipeline(StableDiffusionPipeline):
         self.crop_size = 512
 
         # face_app: FaceAnalysis object
-        self.face_app = FaceAnalysis(name="buffalo_l", root='models/insightface', providers=['CPUExecutionProvider'])
+        self.face_app = FaceAnalysis(name="buffalo_l", root='models/insightface', 
+                                     providers=['CPUExecutionProvider'])
         # The original det_size=(640, 640) is too large and face_app often fails to detect faces.
-        self.face_app.prepare(ctx_id=0, det_size=(512, 512))
+        self.face_app.prepare(ctx_id=0, det_size=(256, 256))
 
         if not os.path.exists(consistentID_weight_path):
             ### Download pretrained models

adaface/face_id_to_ada_prompt.py

@@ -660,7 +660,7 @@ class Arc2Face_ID2AdaPrompt(FaceID2AdaPrompt):
             self.face_app = FaceAnalysis(name='antelopev2', root='models/insightface', 
                                         providers=['CUDAExecutionProvider'],
                                         provider_options=[{"device_id": device_id, 
-                                                           'gpu_mem_limit': 2 * 1024 * 1024 * 1024}])
+                                                           "cudnn_conv_algo_search": "HEURISTIC"}])
             self.face_app.prepare(ctx_id=device_id, det_size=(256, 256))
 
         self.device = device
@@ -806,7 +806,7 @@ class ConsistentID_ID2AdaPrompt(FaceID2AdaPrompt):
             self.face_app = FaceAnalysis(name='buffalo_l', root='models/insightface', 
                                         providers=['CUDAExecutionProvider'],
                                         provider_options=[{"device_id": device_id, 
-                                                           'gpu_mem_limit': 2 * 1024 * 1024 * 1024}])
+                                                           "cudnn_conv_algo_search": "HEURISTIC"}])
             self.face_app.prepare(ctx_id=device_id, det_size=(256, 256))
 
         self.device = device