8000 add --negative-prompt argument to optimize_*.py scripts · chengzeyi/stable-fast@af82bf7 · GitHub
[go: up one dir, main page]
Skip to content

Commit

Permalink
add --negative-prompt argument to optimize_*.py scripts
Browse files Browse the repository at this point in the history
  • Loading branch information
@chengzeyi
chengzeyi committed Jan 31, 2024
1 parent fa9698c commit af82bf7
Show file tree
Hide file tree
Showing 4 changed files with 13 additions and 2 deletions.
4 changes: 3 additions & 1 deletion examples/optimize_instant_id_pipeline.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@
CONTROLNET = 'InstantX/InstantID'
STEPS = 30
PROMPT = 'film noir style, ink sketch|vector, male man, highly detailed, sharp focus, ultra sharpness, monochrome, high contrast, dramatic shadows, 1940s style, mysterious, cinematic'
NEGATIVE_PROMPT = 'ugly, deformed, noisy, blurry, low contrast, realism, photorealistic, vibrant, colorful'
SEED = None
WARMUPS = 3
BATCH = 1
Expand All @@ -17,7 +18,6 @@
CONTROL_IMAGE = None
OUTPUT_IMAGE = None
EXTRA_CALL_KWARGS = '''{
"negative_prompt": "ugly, deformed, noisy, blurry, low contrast, realism, photorealistic, vibrant, colorful",
"controlnet_conditioning_scale": 0.8,
"ip_adapter_scale": 0.8
}'''
Expand Down Expand Up @@ -54,6 +54,7 @@ def parse_args():
parser.add_argument('--controlnet', type=str, default=CONTROLNET)
parser.add_argument('--steps', type=int, default=STEPS)
parser.add_argument('--prompt', type=str, default=PROMPT)
parser.add_argument('--negative-prompt', type=str, default=NEGATIVE_PROMPT)
parser.add_argument('--seed', type=int, default=SEED)
parser.add_argument('--warmups', type=int, default=WARMUPS)
parser.add_argument('--batch', type=int, default=BATCH)
Expand Down Expand Up @@ -254,6 +255,7 @@ def quantize_unet(m):
def get_kwarg_inputs():
kwarg_inputs = dict(
prompt=args.prompt,
negative_prompt=args.negative_prompt,
image_embeds=face_emb,
image=face_kps,
height=height,
Expand Down
5 changes: 4 additions & 1 deletion examples/optimize_lcm_lora.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
CONTROLNET = None
STEPS = 4
PROMPT = 'best quality, realistic, unreal engine, 4K, a beautiful girl'
NEGATIVE_PROMPT = None
SEED = None
WARMUPS = 3
BATCH = 1
Expand Down Expand Up @@ -35,9 +36,10 @@ def parse_args():
parser.add_argument('--custom-pipeline', type=str, default=CUSTOM_PIPELINE)
parser.add_argument('--scheduler', type=str, default=SCHEDULER)
parser.add_argument('--lora', type=str, default=LORA)
parser.add_argument('--controlnet', type=str, default=None)
parser.add_argument('--controlnet', type=str, default=CONTROLNET)
parser.add_argument('--steps', type=int, default=STEPS)
parser.add_argument('--prompt', type=str, default=PROMPT)
parser.add_argument('--negative-prompt', type=str, default=NEGATIVE_PROMPT)
parser.add_argument('--seed', type=int, default=SEED)
parser.add_argument('--warmups', type=int, default=WARMUPS)
parser.add_argument('--batch', type=int, default=BATCH)
Expand Down Expand Up @@ -224,6 +226,7 @@ def quantize_unet(m):
def get_kwarg_inputs():
kwarg_inputs = dict(
prompt=args.prompt,
negative_prompt=args.negative_prompt,
height=height,
width=width,
num_inference_steps=args.steps,
Expand Down
3 changes: 3 additions & 0 deletions examples/optimize_lcm_pipeline.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
CONTROLNET = None
STEPS = 4
PROMPT = 'best quality, realistic, unreal engine, 4K, a beautiful girl'
NEGATIVE_PROMPT = None
SEED = None
WARMUPS = 3
BATCH = 1
Expand Down Expand Up @@ -38,6 +39,7 @@ def parse_args():
parser.add_argument('--controlnet', type=str, default=CONTROLNET)
parser.add_argument('--steps', type=int, default=STEPS)
parser.add_argument('--prompt', type=str, default=PROMPT)
parser.add_argument('--negative-prompt', type=str, default=NEGATIVE_PROMPT)
parser.add_argument('--seed', type=int, default=SEED)
parser.add_argument('--warmups', type=int, default=WARMUPS)
parser.add_argument('--batch', type=int, default=BATCH)
Expand Down Expand Up @@ -224,6 +226,7 @@ def quantize_unet(m):
def get_kwarg_inputs():
kwarg_inputs = dict(
prompt=args.prompt,
negative_prompt=args.negative_prompt,
height=height,
width=width,
num_inference_steps=args.steps,
Expand Down
3 changes: 3 additions & 0 deletions examples/optimize_stable_diffusion_pipeline.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
CONTROLNET = None
STEPS = 30
PROMPT = 'best quality, realistic, unreal engine, 4K, a beautiful girl'
NEGATIVE_PROMPT = None
SEED = None
WARMUPS = 3
BATCH = 1
Expand Down Expand Up @@ -38,6 +39,7 @@ def parse_args():
parser.add_argument('--controlnet', type=str, default=CONTROLNET)
parser.add_argument('--steps', type=int, default=STEPS)
parser.add_argument('--prompt', type=str, default=PROMPT)
parser.add_argument('--negative-prompt', type=str, default=NEGATIVE_PROMPT)
parser.add_argument('--seed', type=int, default=SEED)
parser.add_argument('--warmups', type=int, default=WARMUPS)
parser.add_argument('--batch', type=int, default=BATCH)
Expand Down Expand Up @@ -224,6 +226,7 @@ def quantize_unet(m):
def get_kwarg_inputs():
kwarg_inputs = dict(
prompt=args.prompt,
negative_prompt=args.negative_prompt,
height=height,
width=width,
num_inference_steps=args.steps,
Expand Down

0 comments on commit af82bf7

Please sign in to comment.
0