Spaces:
Configuration error
Configuration 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)
|