Update app/main.py

app/main.py

@@ -40,14 +40,15 @@ async def lifespan(app: FastAPI):
         "briaai/RMBG-1.4", trust_remote_code=True)
     br_model.to(DEVICE)
 
-    sdxl_pipe = StableDiffusionXLPipeline.from_pretrained(
-        base_model_path,
-        torch_dtype=torch.float16,
-        add_watermarker=False,
-    )
-    iv_model = IPAdapterXL(sdxl_pipe, image_encoder_path, ip_ckpt, DEVICE)
-
-    yield {'ti_pipe': ti_pipe, 'ii_pipe': ii_pipe, 'br_model': br_model, 'iv_model': iv_model}
+    # sdxl_pipe = StableDiffusionXLPipeline.from_pretrained(
+    #     base_model_path,
+    #     torch_dtype=torch.float16,
+    #     add_watermarker=False,
+    # )
+    # iv_model = IPAdapterXL(sdxl_pipe, image_encoder_path, ip_ckpt, DEVICE)
+
+    # , 'iv_model': iv_model
+    yield {'ti_pipe': ti_pipe, 'ii_pipe': ii_pipe, 'br_model': br_model}
 
     del ti_pipe
     del ii_pipe