Freak-ppa commited on
Commit
d7ea5d7
1 Parent(s): 84a0854

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -362,7 +362,7 @@ def process(input_fg, prompt, image_width, image_height, num_samples, seed, step
362
  return pytorch2numpy(pixels)
363
 
364
 
365
- @spaces.GPU
366
  @torch.inference_mode()
367
  def process_relight(input_fg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, lowres_denoise, bg_source):
368
  #input_fg, matting = run_rmbg(input_fg)
 
362
  return pytorch2numpy(pixels)
363
 
364
 
365
+ @spaces.GPU(duration=120)
366
  @torch.inference_mode()
367
  def process_relight(input_fg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, lowres_denoise, bg_source):
368
  #input_fg, matting = run_rmbg(input_fg)