Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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)
|