add fidelity tips

app.py

@@ -16,7 +16,6 @@ from diffusers.loaders import LoraLoaderMixin
 import os
 from os.path import join as opj
 
-# run only once
 token = os.getenv("ACCESS_TOKEN")
 os.system(f"python -m pip install git+https://{token}@github.com/logn-2024/StableGarment.git")
 
@@ -24,7 +23,7 @@ from stablegarment.models import AppearanceEncoderModel,ControlNetModel
 from stablegarment.piplines import StableGarmentPipeline,StableGarmentControlNetPipeline
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-torch_dtype = torch.float16 if "cuda"==device else torch.float32
+torch_dtype = torch.float32 if device=="cpu" else torch.float16
 height = 512
 width = 384
 
@@ -37,7 +36,7 @@ garment_encoder = AppearanceEncoderModel.from_pretrained(pretrained_garment_enco
 garment_encoder = garment_encoder.to(device=device,dtype=torch_dtype)
 
 pipeline_t2i = StableGarmentPipeline.from_pretrained(base_model_path, vae=vae, torch_dtype=torch_dtype,).to(device=device) #  variant="fp16"
-# pipeline = StableDiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V4.0_noVAE", vae=vae, torch_dtype=torch_dtype, variant="fp16").to(device=device)
+# pipeline = StableDiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V4.0_noVAE", vae=vae, torch_dtype=torch_dtype).to(device=device)
 pipeline_t2i.scheduler = scheduler
 
 pipeline_tryon = None
@@ -97,13 +96,16 @@ def text2image(prompt,init_image,garment_top,garment_down,style_fidelity=1.):
     garment_top = Image.open(garment_top).resize((width,height))
     garment_top = transforms.CenterCrop((height,width))(transforms.Resize(max(height, width))(garment_top))
 
+    # always enable classifier-free-guidance as it is related to garment
+    cfg = 4 # if prompt else 0 
     garment_images = [garment_top,]
     prompt = [prompt,]
     cloth_prompt = ["",]
     n_prompt = "nsfw, unsaturated, abnormal, unnatural, artifact"
     negative_prompt = [n_prompt]
+    
     images = pipeline_t2i(prompt,negative_prompt=negative_prompt,cloth_prompt=cloth_prompt,height=height,width=width,
-                    num_inference_steps=30,guidance_scale=4,num_images_per_prompt=1,style_fidelity=style_fidelity,
+                    num_inference_steps=30,guidance_scale=cfg,num_images_per_prompt=1,style_fidelity=style_fidelity,
                     garment_encoder=garment_encoder,garment_image=garment_images,).images
     return images[0]
 
@@ -170,8 +172,10 @@ with gr.Blocks(css = ".output-image, .input-image, .image-preview {height: 400px
             with gr.Row():
                 t2i_only = gr.Checkbox(label="t2i with garment", info="Only text and garment.", elem_id="t2i_switch", value=True, interactive=False,)
                 run_button = gr.Button(value="Run")
-                style_fidelity = gr.Slider(0, 1, value=1, label="fidelity(for t2i)") # , info=""
                 t2i_only.change(fn=set_mode,inputs=[t2i_only,init_image,prompt],outputs=[init_image,prompt,])
+            with gr.Accordion("advance options", open=False):
+                gr.Markdown("Garment fidelity control(Tune down it to reduce white edge).")
+                style_fidelity = gr.Slider(0, 1, value=1, label="fidelity(only for t2i)") # , info=""
         with gr.Column():
             gallery = gr.Image()
             run_button.click(fn=infer,