Update app.py

app.py

@@ -32,7 +32,7 @@ start_time = time.time()
 ####################### Setup Model
 model_id = "stabilityai/stable-diffusion-xl-base-1.0"
 lcm_lora_id = "latent-consistency/lcm-lora-sdxl"
-pipe = SDEmb.from_pretrained(model_id, variant="fp16", low_mem_cpu_usage=True)
+pipe = SDEmb.from_pretrained(model_id, variant="fp16", low_cpu_mem_usage=True, device_map="auto")
 pipe.load_lora_weights(lcm_lora_id)
 pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
 pipe.to(device='cuda', dtype=torch.float16)