Update inference_manager.py

inference_manager.py

@@ -476,6 +476,8 @@ class ModelManager:
 
     @spaces.GPU(duration=40)
     def generate_with_faceid(self, model_id, inference_params, progress=gr.Progress(track_tqdm=True)):
+        # Clear GPU memory
+        torch.cuda.empty_cache()
         model = self.models.get(model_id)
         if not model:
             raise Exception(f"invalid model_id {model_id}")
@@ -507,7 +509,6 @@ class ModelManager:
         faceid_all_embeds = []
         for image in images:
             face = image#cv2.imread(image)
-            print(image)
             faces = self.app.get(face)
             faceid_embed = torch.from_numpy(faces[0].normed_embedding).unsqueeze(0)
             faceid_all_embeds.append(faceid_embed)