tonyassi commited on
Commit
11e01c6
1 Parent(s): 300e174

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -19,7 +19,7 @@ inpaint_pipeline = AutoPipelineForInpainting.from_pretrained("diffusers/stable-d
19
  inpaint_pipeline.load_ip_adapter("h94/IP-Adapter", subfolder="sdxl_models", weight_name="ip-adapter_sdxl.bin")
20
  inpaint_pipeline.set_ip_adapter_scale(0.6)
21
 
22
- #@spaces.GPU(enable_queue=True)
23
  def text_to_image(ip, prompt, neg_prompt, width, height, ip_scale, strength, guidance, steps):
24
  text_pipeline.to("cuda")
25
 
@@ -40,7 +40,7 @@ def text_to_image(ip, prompt, neg_prompt, width, height, ip_scale, strength, gui
40
 
41
  return images[0]
42
 
43
- #@spaces.GPU(enable_queue=True)
44
  def image_to_image(ip, image, prompt, neg_prompt, width, height, ip_scale, strength, guidance, steps):
45
  image_pipeline.to("cuda")
46
 
@@ -63,7 +63,7 @@ def image_to_image(ip, image, prompt, neg_prompt, width, height, ip_scale, stren
63
 
64
  return images[0]
65
 
66
- #@spaces.GPU(enable_queue=True)
67
  def inpaint(ip, image_editor, prompt, neg_prompt, width, height, ip_scale, strength, guidance, steps):
68
  inpaint_pipeline.to("cuda")
69
  print(image_editor)
 
19
  inpaint_pipeline.load_ip_adapter("h94/IP-Adapter", subfolder="sdxl_models", weight_name="ip-adapter_sdxl.bin")
20
  inpaint_pipeline.set_ip_adapter_scale(0.6)
21
 
22
+ @spaces.GPU(enable_queue=True)
23
  def text_to_image(ip, prompt, neg_prompt, width, height, ip_scale, strength, guidance, steps):
24
  text_pipeline.to("cuda")
25
 
 
40
 
41
  return images[0]
42
 
43
+ @spaces.GPU(enable_queue=True)
44
  def image_to_image(ip, image, prompt, neg_prompt, width, height, ip_scale, strength, guidance, steps):
45
  image_pipeline.to("cuda")
46
 
 
63
 
64
  return images[0]
65
 
66
+ @spaces.GPU(enable_queue=True)
67
  def inpaint(ip, image_editor, prompt, neg_prompt, width, height, ip_scale, strength, guidance, steps):
68
  inpaint_pipeline.to("cuda")
69
  print(image_editor)