Shitao commited on
Commit
bc7d1d7
1 Parent(s): 65e7ed8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ pipe = OmniGenPipeline.from_pretrained(
10
  "Shitao/OmniGen-v1"
11
  )
12
 
13
- @spaces.GPU(duration=160)
14
  def generate_image(text, img1, img2, img3, height, width, guidance_scale, img_guidance_scale, inference_steps, seed, separate_cfg_infer, offload_model,
15
  use_input_image_size_as_output, max_input_image_size, randomize_seed):
16
  input_images = [img1, img2, img3]
 
10
  "Shitao/OmniGen-v1"
11
  )
12
 
13
+ @spaces.GPU(duration=200)
14
  def generate_image(text, img1, img2, img3, height, width, guidance_scale, img_guidance_scale, inference_steps, seed, separate_cfg_infer, offload_model,
15
  use_input_image_size_as_output, max_input_image_size, randomize_seed):
16
  input_images = [img1, img2, img3]