multimodalart HF staff commited on
Commit
debbd96
1 Parent(s): 21f9f22

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
32
 
33
  ip_model = IPAdapterFaceID(pipe, ip_ckpt, device)
34
 
35
- @spaces.GPU
36
  def generate_image(images, prompt, negative_prompt, progress=gr.Progress(track_tqdm=True)):
37
  pipe.to(device)
38
  app = FaceAnalysis(name="buffalo_l", providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
 
32
 
33
  ip_model = IPAdapterFaceID(pipe, ip_ckpt, device)
34
 
35
+ @spaces.GPU(enable_queue=True)
36
  def generate_image(images, prompt, negative_prompt, progress=gr.Progress(track_tqdm=True)):
37
  pipe.to(device)
38
  app = FaceAnalysis(name="buffalo_l", providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])