Update app.py

app.py

@@ -6,7 +6,7 @@ import gradio as gr
 import torch
 import spaces
 
-vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse")
+vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse", torch_dtype=torch.float16)
 #safety_checker=StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker"),
 #feature_extractor=CLIPFeatureExtractor.from_pretrained("openai/clip-vit-base-patch32")
 pretrain_model = "SG161222/Realistic_Vision_V5.1_noVAE"
@@ -21,8 +21,8 @@ pipe = StableDiffusionPipeline.from_pretrained(pretrain_model, scheduler = sched
 apply_hidiffusion(pipe)
 
 @spaces.GPU
-def run_hidiffusion(prompt, negative_prompt):
-    return pipe(prompt, guidance_scale=7.5, height=1024, width=1024, eta=1.0, negative_prompt=negative_prompt).images[0]
+def run_hidiffusion(prompt, negative_prompt, progress=gr.Progress(track_tqdm=True)):
+    return pipe(prompt, guidance_scale=7.5, height=1024, width=1024, eta=1.0, negative_prompt=negative_prompt, num_inference_steps=25).images[0]
         
 with gr.Blocks() as demo:
     prompt = gr.Textbox()