mirror of
https://github.com/nod-ai/SHARK-Studio.git
synced 2026-01-08 05:24:00 -05:00
99 lines
2.4 KiB
Python
99 lines
2.4 KiB
Python
import argparse
|
|
|
|
p = argparse.ArgumentParser(
|
|
description=__doc__, formatter_class=argparse.ArgumentDefaultsHelpFormatter
|
|
)
|
|
|
|
##############################################################################
|
|
### Stable Diffusion Params
|
|
##############################################################################
|
|
|
|
p.add_argument(
|
|
"--prompts",
|
|
nargs="+",
|
|
default=["cyberpunk forest by Salvador Dali"],
|
|
help="text of which images to be generated.",
|
|
)
|
|
|
|
p.add_argument(
|
|
"--negative-prompts",
|
|
nargs="+",
|
|
default=[""],
|
|
help="text you don't want to see in the generated image.",
|
|
)
|
|
|
|
p.add_argument(
|
|
"--steps",
|
|
type=int,
|
|
default=50,
|
|
help="the no. of steps to do the sampling.",
|
|
)
|
|
|
|
p.add_argument(
|
|
"--seed",
|
|
type=int,
|
|
default=42,
|
|
help="the seed to use.",
|
|
)
|
|
|
|
p.add_argument(
|
|
"--guidance_scale",
|
|
type=float,
|
|
default=7.5,
|
|
help="the value to be used for guidance scaling.",
|
|
)
|
|
|
|
##############################################################################
|
|
### Model Config and Usage Params
|
|
##############################################################################
|
|
|
|
p.add_argument(
|
|
"--device", type=str, default="vulkan", help="device to run the model."
|
|
)
|
|
|
|
p.add_argument(
|
|
"--precision", type=str, default="fp16", help="precision to run the model."
|
|
)
|
|
|
|
p.add_argument(
|
|
"--import_mlir",
|
|
default=False,
|
|
action=argparse.BooleanOptionalAction,
|
|
help="imports the model from torch module to amdshark_module otherwise downloads the model from amdshark_tank.",
|
|
)
|
|
|
|
p.add_argument(
|
|
"--load_vmfb",
|
|
default=True,
|
|
action=argparse.BooleanOptionalAction,
|
|
help="attempts to load the model from a precompiled flatbuffer and compiles + saves it if not found.",
|
|
)
|
|
|
|
p.add_argument(
|
|
"--save_vmfb",
|
|
default=False,
|
|
action=argparse.BooleanOptionalAction,
|
|
help="saves the compiled flatbuffer to the local directory",
|
|
)
|
|
|
|
##############################################################################
|
|
### IREE - Vulkan supported flags
|
|
##############################################################################
|
|
|
|
p.add_argument(
|
|
"--iree-vulkan-target-triple",
|
|
type=str,
|
|
default="",
|
|
help="Specify target triple for vulkan",
|
|
)
|
|
|
|
p.add_argument(
|
|
"--vulkan_debug_utils",
|
|
default=False,
|
|
action=argparse.BooleanOptionalAction,
|
|
help="Profiles vulkan device and collects the .rdc info",
|
|
)
|
|
|
|
|
|
args = p.parse_args()
|