Update app.py

app.py

@@ -81,17 +81,16 @@ NUM_IMAGES_PER_PROMPT = 1
 pipe = PixArtAlphaPipeline.from_pretrained(
         "PixArt-alpha/PixArt-LCM-XL-2-1024-MS",
         torch_dtype=torch.float16,
-        use_safetensors=True,
-    ).to(device)
+        use_safetensors=True).to(device)
+
+pipe.vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=torch.float16)
+pipe.text_encoder.to_bettertransformer()
 
 def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
     p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
     if not negative:
         negative = ""
     return p.replace("{prompt}", positive), n + negative
-        
-    # speed-up T5
-    pipe.text_encoder.to_bettertransformer()
 
     if USE_TORCH_COMPILE:
         pipe.transformer = torch.compile(pipe.transformer, mode="reduce-overhead", fullgraph=True)
@@ -149,6 +148,7 @@ def generate(
 examples = [
     "A Monkey with a happy face in the Sahara desert.",
     "Eiffel Tower was Made up of ICE.",
+    "photo of 8k ultra realistic harbour, nreal engine 5, port, boats, sunset, beautiful light, full of colour, cinematic lighting, battered, trending on artstation, 4k, hyperrealistic, focused, extreme details",
     "Color photo of a corgi made of transparent glass, standing on the riverside in Yosemite National Park.",
     "A close-up photo of a woman. She wore a blue coat with a gray dress underneath and has blue eyes.",
     "A litter of golden retriever puppies playing in the snow. Their heads pop out of the snow, covered in.",
@@ -225,7 +225,7 @@ with gr.Blocks(css=css) as demo:
                 minimum=4,
                 maximum=20,
                 step=1,
-                value=4,
+                value=8,
             )
 
     gr.Examples(