Update app.py

add weight_type

app.py

@@ -20,6 +20,7 @@ if torch.cuda.is_available():
     # pipe.vae = AutoencoderKL.from_pretrained("IDKiro/sdxs-512-0.9/vae_large")     # use original VAE
     pipe.to("cuda")
 else:
+    weight_type = torch.float16
     pipe = StableDiffusionPipeline.from_pretrained(repo, torch_dtype=weight_type)
 
 # Function