Update app.py

app.py

@@ -20,8 +20,8 @@ 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)
-    pipe.vae = StableDiffusionPipeline.from_pretrained(vae_path).vae
+    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):