Update app.py

app.py

@@ -79,14 +79,14 @@ NUM_IMAGES_PER_PROMPT = 3
 
 if torch.cuda.is_available():
     pipe = DiffusionPipeline.from_pretrained(
-        "fluently/Fluently-XL-Final",
+        "SG161222/RealVisXL_V4.0",
         torch_dtype=torch.float16,
         use_safetensors=True,
         add_watermarker=False,
         variant="fp16",
     )
     pipe2 = DiffusionPipeline.from_pretrained(
-        "SG161222/RealVisXL_V4.0",
+        "SG161222/RealVisXL_V3.0_Turbo",
         torch_dtype=torch.float16,
         use_safetensors=True,
         add_watermarker=False,
@@ -101,8 +101,8 @@ if torch.cuda.is_available():
         print("Loaded on Device!")
 
     if USE_TORCH_COMPILE:
-        pipe.unet = torch.compile(pipe.unet)
-        pipe2.unet = torch.compile(pipe2.unet)
+        pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
+        pipe2.unet = torch.compile(pipe2.unet, mode="reduce-overhead", fullgraph=True)
         print("Model Compiled!")
 
 
@@ -122,14 +122,14 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
 def generate(
     prompt: str,
     negative_prompt: str = "",
-    use_negative_prompt: bool = True,
+    use_negative_prompt: bool = False,
     style: str = DEFAULT_STYLE,
-    seed: int = 999,
+    seed: int = 0,
     width: int = 896,
     height: int = 1152,
     guidance_scale: float = 3,
     randomize_seed: bool = False,
-    use_resolution_binning: bool = False,
+    use_resolution_binning: bool = True,
     progress=gr.Progress(track_tqdm=True),
 ):
 
@@ -143,7 +143,7 @@ def generate(
         "width": width,
         "height": height,
         "guidance_scale": guidance_scale,
-        "num_inference_steps": 30,
+        "num_inference_steps": 25,
         "generator": generator,
         "num_images_per_prompt": NUM_IMAGES_PER_PROMPT,
         "use_resolution_binning": use_resolution_binning,
@@ -158,7 +158,9 @@ def generate(
 
 examples = [
     (
-        "Photograph of ur mom, and she's hella hot"
+        "college life of 21 year old, depth of field, bokeh, shallow"
+        " focus, minimalism, fujifilm xh2s with Canon EF lens, cinematic --ar 85:128"
+        " --v 6.0 --style raw"
     ),
 ]
 
@@ -284,4 +286,4 @@ with gr.Blocks(css=css, theme="rawrsor1/Everforest") as demo:
     )
 
 if __name__ == "__main__":
-    demo.queue(max_size=20).launch()
+    demo.queue(max_size=20).launch()