Update app.py

app.py

@@ -9,7 +9,7 @@ import numpy as np
 from PIL import Image
 import spaces
 import torch
-from diffusers import StableDiffusionXLPipeline
+from diffusers import StableDiffusionXLPipeline, KDPM2AncestralDiscreteScheduler, AutoencoderKL
 
 DESCRIPTION = """
 # Proteus V0.1
@@ -37,6 +37,11 @@ if torch.cuda.is_available():
     if ENABLE_CPU_OFFLOAD:
         pipe.enable_model_cpu_offload()
     else:
+        vae = AutoencoderKL.from_pretrained(
+            "madebyollin/sdxl-vae-fp16-fix", 
+            torch_dtype=torch.float16
+        )
+        pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)
         pipe.to(device)    
         print("Loaded on Device!")
     pipe.load_lora_weights("stabilityai/stable-diffusion-xl-base-1.0", weight_name="sd_xl_offset_example-lora_1.0.safetensors")
@@ -168,7 +173,7 @@ with gr.Blocks(title="Proteus V0.1", css=css) as demo:
                 minimum=0.1,
                 maximum=20,
                 step=0.1,
-                value=8.0,
+                value=7.0,
             )
 
     gr.Examples(