Fucius commited on
Commit
4f87b73
1 Parent(s): 9db7d3b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=120)
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)
@@ -657,7 +657,7 @@ def main(device, segment_type):
657
 
658
  def parse_args():
659
  parser = argparse.ArgumentParser('', add_help=False)
660
- parser.add_argument('--pretrained_model', default='stablediffusionapi/realism-engine-sdxl-v30', type=str)
661
  parser.add_argument('--controlnet_path', default='/home/user/app/checkpoint/InstantID/ControlNetModel', type=str)
662
  parser.add_argument('--face_adapter_path', default='/home/user/app/checkpoint/InstantID/ip-adapter.bin', type=str)
663
  parser.add_argument('--openpose_checkpoint', default='/home/user/app/checkpoint/controlnet-openpose-sdxl-1.0', type=str)
 
427
  image = Image.fromarray((image * 255.0).clip(0, 255).astype(np.uint8))
428
  return image
429
 
430
+ @spaces.GPU(duration=140)
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)
 
657
 
658
  def parse_args():
659
  parser = argparse.ArgumentParser('', add_help=False)
660
+ parser.add_argument('--pretrained_model', default='wangqixun/YamerMIX_v8', type=str)
661
  parser.add_argument('--controlnet_path', default='/home/user/app/checkpoint/InstantID/ControlNetModel', type=str)
662
  parser.add_argument('--face_adapter_path', default='/home/user/app/checkpoint/InstantID/ip-adapter.bin', type=str)
663
  parser.add_argument('--openpose_checkpoint', default='/home/user/app/checkpoint/controlnet-openpose-sdxl-1.0', type=str)