Update app.py

app.py

@@ -110,21 +110,40 @@ face_adapter = f'/data/checkpoints/ip-adapter.bin'
 controlnet_path = f'/data/checkpoints/ControlNetModel'
 
 # load IdentityNet
+st = time.time()
 identitynet = ControlNetModel.from_pretrained(controlnet_path, torch_dtype=torch.float16)
 zoedepthnet = ControlNetModel.from_pretrained("diffusers/controlnet-zoe-depth-sdxl-1.0",torch_dtype=torch.float16)
+et = time.time()
+elapsed_time = et - st
+print('Loading ControlNet took: ', elapsed_time, 'seconds')
+st = time.time()
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
+et = time.time()
+elapsed_time = et - st
+print('Loading VAE took: ', elapsed_time, 'seconds')
+st = time.time()
 pipe = StableDiffusionXLInstantIDImg2ImgPipeline.from_pretrained("rubbrband/albedobaseXL_v21",
                                                                  vae=vae,
                                                                  controlnet=[identitynet, zoedepthnet],
                                                                  torch_dtype=torch.float16)
-
-compel = Compel(tokenizer=[pipe.tokenizer, pipe.tokenizer_2] , text_encoder=[pipe.text_encoder, pipe.text_encoder_2], returned_embeddings_type=ReturnedEmbeddingsType.PENULTIMATE_HIDDEN_STATES_NON_NORMALIZED, requires_pooled=[False, True])
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config, use_karras_sigmas=True)
 pipe.load_ip_adapter_instantid(face_adapter)
 pipe.set_ip_adapter_scale(0.8)
+et = time.time()
+elapsed_time = et - st
+print('Loading pipeline took: ', elapsed_time, 'seconds')
+st = time.time()
+compel = Compel(tokenizer=[pipe.tokenizer, pipe.tokenizer_2] , text_encoder=[pipe.text_encoder, pipe.text_encoder_2], returned_embeddings_type=ReturnedEmbeddingsType.PENULTIMATE_HIDDEN_STATES_NON_NORMALIZED, requires_pooled=[False, True])
+et = time.time()
+elapsed_time = et - st
+print('Loading Compel took: ', elapsed_time, 'seconds')
+
+st = time.time()
 zoe = ZoeDetector.from_pretrained("lllyasviel/Annotators")
+et = time.time()
+elapsed_time = et - st
+print('Loading Zoe took: ', elapsed_time, 'seconds')
 zoe.to(device)
-
 pipe.to(device)
 
 last_lora = ""