Update app.py

app.py

@@ -59,12 +59,13 @@ def generate(
     url = '',
 ):
     if torch.cuda.is_available():
-
-        if use_vae:  
-            vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)
             
         if not use_img2img:
-            pipe = DiffusionPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16)
+            pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
+
+                if use_vae:  
+                    vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)
+                    pipe = DiffusionPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16)
             
         if use_img2img:
             pipe = AutoPipelineForImage2Image.from_pretrained(model, vae=vae, torch_dtype=torch.float16)