Update app.py

app.py

@@ -15,7 +15,10 @@ from small_256_model import UNet as small_UNet
 from big_1024_model import UNet as big_UNet
 from CLIP import load as load_clip,load_vae,encode_prompt
 from rich import print as rp
+from diffusers import AutoencoderKL
 
+#url = "https://huggingface.co/stabilityai/sd-vae-ft-mse-original/blob/main/vae-ft-mse-840000-ema-pruned.safetensors"  # can also be a local file
+#model = AutoencoderKL.from_single_file(url)
 # Device configuration
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 big = False if device == torch.device('cpu') else True