Update app.py

app.py

@@ -13,14 +13,10 @@ model_paths = {
     "stabilityai/stable-diffusion-2-1": "stabilityai/stable-diffusion-2-1"
 }
 
-# VAE 경로
-vae_path = "stabilityai/sd-vae-ft-mse"
-
-# 모델 및 VAE를 로드합니다.
+# 모델을 로드합니다.
 pipes = {}
 for name, path in model_paths.items():
     pipe = StableDiffusionPipeline.from_pretrained(path, torch_dtype=torch.float16, low_cpu_mem_usage=True)
-    pipe.vae = StableDiffusionPipeline.from_pretrained(vae_path, low_cpu_mem_usage=True).vae
     pipes[name] = pipe
 
 def generate_image(prompt, negative_prompt, model_name, steps, sampler, cfg_scale, width, height, seed):