Update app.py

app.py

@@ -76,10 +76,14 @@ def get_ControlNetXS(base_model, cnxs_path, device, size_ratio=0.125, weight_dty
     
     return ControlNetXS
 
+print('Get ControlNetXS...')
 ControlNetXS = get_ControlNetXS(base_model, cnxs_path, device, size_ratio=0.125, weight_dtype=weight_dtype)
+
 vae = AutoencoderKL.from_pretrained(vae_path)
+
+print('Get Pipeline...')
 pipe = StableDiffusionXLInstantIDXSPipeline.from_pretrained(
-    pretrained_model_name_or_path,
+    base_model,
     vae=vae,
     unet=ControlNetXS,
     controlnet=None,
@@ -87,11 +91,13 @@ pipe = StableDiffusionXLInstantIDXSPipeline.from_pretrained(
 )
 
 pipe.cuda(device=device, dtype=weight_dtype, use_xformers=True)
+
+print('Load IP-Adapter')
 pipe.load_ip_adapter(image_proj_path, cross_attn_path)
 
 pipe.scheduler = diffusers.EulerDiscreteScheduler.from_config(pipe.scheduler.config)
 pipe.unet.config.ctrl_learn_time_embedding = True
-pipe = pipe.to(args.device)
+pipe = pipe.to(device)