Update app.py

app.py

@@ -42,7 +42,7 @@ weight_dtype = torch.float16 if str(device).__contains__("cuda") else torch.floa
 
 base_model = 'frankjoshua/realvisxlV40_v40Bakedvae'
 vae_path = 'madebyollin/sdxl-vae-fp16-fix'
-ckpt = 'RED-AIGC/InstantID-XS'
+# ckpt = 'RED-AIGC/InstantID-XS'
 
 image_proj_path = "./ckpt/image_proj.bin"
 cnxs_path =  "./ckpt/controlnetxs.bin"
@@ -90,9 +90,10 @@ pipe = StableDiffusionXLInstantIDXSPipeline.from_pretrained(
     torch_dtype=weight_dtype,
 )
 
-pipe.cuda(device=device, dtype=weight_dtype, use_xformers=True)
+# pipe.cuda(device=device, dtype=weight_dtype, use_xformers=True)
+pipe.cuda(device=device, dtype=weight_dtype, use_xformers=False)
 
-print('Load IP-Adapter')
+print('Load IP-Adapter...')
 pipe.load_ip_adapter(image_proj_path, cross_attn_path)
 
 pipe.scheduler = diffusers.EulerDiscreteScheduler.from_config(pipe.scheduler.config)
@@ -317,20 +318,15 @@ with gr.Blocks(css=css) as demo:
         with gr.Column():
             with gr.Row(equal_height=True):
                 # upload face image
-                face_file = gr.Image(
-                    label="Upload a photo of your face", type="filepath"
-                )
+                face_file = gr.Image(label="Upload a photo of your face", type="filepath")
                 # optional: upload a reference pose image
-                pose_file = gr.Image(
-                    label="Upload a reference pose image (Optional)",
-                    type="filepath",
-                )
+                pose_file = gr.Image(label="Upload a reference pose image (Optional)",type="filepath",)
 
             # prompt
             prompt = gr.Textbox(
                 label="Prompt",
                 info="Give simple prompt is enough to achieve good face fidelity",
-                placeholder="A photo of a person",
+                placeholder="a woman, (looking at the viewer), portrait, daily wear, 8K texture, realistic, symmetrical hyperdetailed texture, masterpiece, enhanced details, (eye highlight:2),   perfect composition, natural lighting, best quality, authentic, natural posture",
                 value="",
             )