Update inference_manager.py

inference_manager.py

@@ -196,7 +196,7 @@ class InferenceManager:
             pipe = StableDiffusionPipeline.from_pretrained(ckpt_dir, vae=vae, torch_dtype=torch.bfloat16, use_safetensors=True)
         else:
             use_vae = cfg.get("vae", "")
-            if not use_vae or True:#!TEST! default vae for test
+            if not use_vae:
                 vae = AutoencoderKL.from_pretrained(os.path.join(ckpt_dir, "vae"), torch_dtype=torch.bfloat16)
             elif use_vae == "tae":
                 vae = AutoencoderTiny.from_pretrained("madebyollin/taesdxl", torch_dtype=torch.bfloat16)
@@ -510,7 +510,7 @@ class ModelManager:
         print("extracting face...")
         faceid_all_embeds = []
         for image in images:
-            face = image#cv2.imread(image)
+            face = cv2.imread(image) ##here accepts image path
             faces = self.app.get(face)
             faceid_embed = torch.from_numpy(faces[0].normed_embedding).unsqueeze(0)
             faceid_all_embeds.append(faceid_embed)