Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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=
|
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='
|
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)
|