Update app.py

app.py

@@ -22,13 +22,14 @@ ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 if torch.cuda.is_available():
-    pipe = StableDiffusionXLPipeline.from_pretrained(
+    vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
+    pipe = DiffusionPipeline.from_pretrained(
         "dataautogpt3/OpenDalleV1.1",
+        vae=vae,
         torch_dtype=torch.float16,
         use_safetensors=True,
-        add_watermarker=False
+        variant="fp16",
     )
-    pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
     pipe.to("cuda")
 
 
@@ -154,9 +155,9 @@ with gr.Blocks(css=css) as demo:
             num_inference_steps = gr.Slider(
                 label="Number of inference steps",
                 minimum=1,
-                maximum=15,
+                maximum=50,
                 step=1,
-                value=8,
+                value=30,
             )
 
     gr.Examples(