Fucius commited on
Commit
ce6a1ec
·
verified ·
1 Parent(s): f24b7ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -427,7 +427,7 @@ def main(device, segment_type):
427
  image = Image.fromarray((image * 255.0).clip(0, 255).astype(np.uint8))
428
  return image
429
 
430
- @spaces.GPU(duration=150)
431
  def generate_image(prompt1, negative_prompt, reference_1, reference_2, resolution, local_prompt1, local_prompt2, seed, style, identitynet_strength_ratio, adapter_strength_ratio, condition, condition_img, controlnet_ratio, cfg_scale):
432
  identitynet_strength_ratio = float(identitynet_strength_ratio)
433
  adapter_strength_ratio = float(adapter_strength_ratio)
 
427
  image = Image.fromarray((image * 255.0).clip(0, 255).astype(np.uint8))
428
  return image
429
 
430
+ @spaces.GPU(duration=180)
431
  def generate_image(prompt1, negative_prompt, reference_1, reference_2, resolution, local_prompt1, local_prompt2, seed, style, identitynet_strength_ratio, adapter_strength_ratio, condition, condition_img, controlnet_ratio, cfg_scale):
432
  identitynet_strength_ratio = float(identitynet_strength_ratio)
433
  adapter_strength_ratio = float(adapter_strength_ratio)