Skip reloading face_app when on hf space

adaface/face_id_to_ada_prompt.py

@@ -655,8 +655,13 @@ class Arc2Face_ID2AdaPrompt(FaceID2AdaPrompt):
         # No need to reload face_app on the same device.
         if device == self.device:
             return
+        self.device = device
         
-        if str(device) == 'cpu' or self.is_on_hf_space:
+        if self.is_on_hf_space and self.face_app is not None:
+            print(f'On HF space. Arc2Face Face encoder already loaded on cpu.')
+            return 
+        
+        if str(device) == 'cpu':
             self.face_app = FaceAnalysis(name='antelopev2', root='models/insightface', 
                                          providers=['CPUExecutionProvider'])
             self.face_app.prepare(ctx_id=0, det_size=(512, 512))
@@ -668,7 +673,6 @@ class Arc2Face_ID2AdaPrompt(FaceID2AdaPrompt):
                                                             'cudnn_conv_algo_search': 'HEURISTIC'}])
             self.face_app.prepare(ctx_id=device_id, det_size=(512, 512))
 
-        self.device = device
         print(f'Arc2Face Face encoder reloaded on {device}.')
         return 
         
@@ -801,6 +805,11 @@ class ConsistentID_ID2AdaPrompt(FaceID2AdaPrompt):
         # No need to reload face_app on the same device.
         if device == self.device:
             return
+        self.device = device
+
+        if self.is_on_hf_space and self.face_app is not None:
+            print(f'On HF space. Arc2Face Face encoder already loaded on cpu.')
+            return 
 
         if str(device) == 'cpu' or self.is_on_hf_space:
             self.face_app = FaceAnalysis(name='buffalo_l', root='models/insightface', 
@@ -814,7 +823,6 @@ class ConsistentID_ID2AdaPrompt(FaceID2AdaPrompt):
                                                             'cudnn_conv_algo_search': 'HEURISTIC'}])
             self.face_app.prepare(ctx_id=device_id, det_size=(512, 512))
 
-        self.device = device
         self.pipe.face_app = self.face_app
         print(f'ConsistentID Face encoder reloaded on {device}.')