Ashoka74 commited on
Commit
21dbed4
ยท
verified ยท
1 Parent(s): 1c2a234

Update app_merged.py

Browse files
Files changed (1) hide show
  1. app_merged.py +3 -2
app_merged.py CHANGED
@@ -351,7 +351,8 @@ def prepare_image_and_mask(image, width, height, overlap_percentage, resize_opti
351
 
352
  return background, mask
353
 
354
- @spaces.GPU
 
355
  def inpaint(image, width, height, overlap_percentage, num_inference_steps, resize_option, custom_resize_percentage, prompt_input, alignment, overlap_left, overlap_right, overlap_top, overlap_bottom, progress=gr.Progress(track_tqdm=True)):
356
 
357
  background, mask = prepare_image_and_mask(image, width, height, overlap_percentage, resize_option, custom_resize_percentage, alignment, overlap_left, overlap_right, overlap_top, overlap_bottom)
@@ -609,7 +610,7 @@ def infer(
609
  seed=42,
610
  randomize_seed=False,
611
  guidance_scale=3.0,
612
- num_inference_steps=50,
613
  reference_conditioning_scale=1.0,
614
  negative_prompt="watermark, ugly, deformed, noisy, blurry, low contrast",
615
  progress=gr.Progress(track_tqdm=True),
 
351
 
352
  return background, mask
353
 
354
+ @spaces.GPU(duration=60)
355
+ @torch.inference_mode()
356
  def inpaint(image, width, height, overlap_percentage, num_inference_steps, resize_option, custom_resize_percentage, prompt_input, alignment, overlap_left, overlap_right, overlap_top, overlap_bottom, progress=gr.Progress(track_tqdm=True)):
357
 
358
  background, mask = prepare_image_and_mask(image, width, height, overlap_percentage, resize_option, custom_resize_percentage, alignment, overlap_left, overlap_right, overlap_top, overlap_bottom)
 
610
  seed=42,
611
  randomize_seed=False,
612
  guidance_scale=3.0,
613
+ num_inference_steps=30,
614
  reference_conditioning_scale=1.0,
615
  negative_prompt="watermark, ugly, deformed, noisy, blurry, low contrast",
616
  progress=gr.Progress(track_tqdm=True),