# Copyright (c) 2023, Teriks
#
# dgenerate is distributed under the following BSD 3-Clause License
#
# Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:
#
# 1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.
#
# 2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in
# the documentation and/or other materials provided with the distribution.
#
# 3. Neither the name of the copyright holder nor the names of its contributors may be used to endorse or promote products derived
# from this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
# HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
# ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
import argparse
import typing
from argparse import Action
import diffusers.schedulers
import dgenerate
import dgenerate.mediaoutput as _mediaoutput
import dgenerate.memory as _memory
import dgenerate.messages as _messages
import dgenerate.pipelinewrapper as _pipelinewrapper
import dgenerate.prompt as _prompt
import dgenerate.textprocessing as _textprocessing
import dgenerate.types as _types
_SUPPORTED_MODEL_TYPES_PRETTY = \
_textprocessing.oxford_comma(_pipelinewrapper.supported_model_type_strings(), 'or')
_SUPPORTED_ANIMATION_OUTPUT_FORMATS_PRETTY = \
_textprocessing.oxford_comma(_mediaoutput.supported_animation_writer_formats(), 'or')
_SUPPORTED_DATA_TYPES_PRETTY = \
_textprocessing.oxford_comma(_pipelinewrapper.supported_data_type_strings(), 'or')
parser = argparse.ArgumentParser(
prog='dgenerate', exit_on_error=False,
description="""Stable diffusion batch image generation tool with
support for video / gif / webp animation transcoding.""")
def _model_type(val):
val = val.lower()
if val not in _pipelinewrapper.supported_model_type_strings():
raise argparse.ArgumentTypeError(
f'Must be one of: {_SUPPORTED_MODEL_TYPES_PRETTY}. Unknown value: {val}')
return _pipelinewrapper.get_model_type_enum(val)
def _type_dtype(dtype):
dtype = dtype.lower()
supported_dtypes = _pipelinewrapper.supported_data_type_strings()
if dtype not in supported_dtypes:
raise argparse.ArgumentTypeError(f'Must be {_textprocessing.oxford_comma(supported_dtypes, "or")}.')
else:
return _pipelinewrapper.get_data_type_enum(dtype)
def _type_prompts(prompt):
try:
return _prompt.Prompt.parse(prompt)
except ValueError as e:
raise argparse.ArgumentTypeError(
f'Prompt parse error: {e}')
def _type_clip_skip(val):
try:
val = int(val)
except ValueError:
raise argparse.ArgumentTypeError('Must be an integer')
if val < 0:
raise argparse.ArgumentTypeError('Must be greater than or equal to 0')
return val
def _type_inference_steps(val):
try:
val = int(val)
except ValueError:
raise argparse.ArgumentTypeError('Must be an integer')
if val <= 0:
raise argparse.ArgumentTypeError('Must be greater than 0')
return val
def _type_guidance_scale(val):
try:
val = float(val)
except ValueError:
raise argparse.ArgumentTypeError('Must be a floating point number')
if val < 0:
raise argparse.ArgumentTypeError('Must be greater than or equal to 0')
return val
def _type_size(size):
if size is None:
return None
try:
r = size.lower().split('x')
if len(r) < 2:
x, y = int(r[0]), int(r[0])
else:
x, y = int(r[0]), int(r[1])
except ValueError:
raise argparse.ArgumentTypeError('Dimensions must be integer values.')
if x * y < 1:
raise argparse.ArgumentTypeError('Dimensions must have a product of at least 1.')
return x, y
def _type_output_size(size):
x, y = _type_size(size)
if x % 8 != 0:
raise argparse.ArgumentTypeError('Output X dimension must be divisible by 8.')
if y % 8 != 0:
raise argparse.ArgumentTypeError('Output Y dimension must be divisible by 8.')
return x, y
def _type_image_coordinate(coord):
if coord is None:
return 0, 0
r = coord.split(',')
try:
return int(r[0]), int(r[1])
except ValueError:
raise argparse.ArgumentTypeError('Coordinates must be integer values.')
def _type_sdxl_high_noise_fractions(val):
try:
val = float(val)
except ValueError:
raise argparse.ArgumentTypeError('Must be a floating point number')
if val <= 0:
raise argparse.ArgumentTypeError('Must be greater than 0')
return val
def _type_device(device):
try:
if not _pipelinewrapper.is_valid_device_string(device):
raise argparse.ArgumentTypeError(f'Must be cuda or cpu. Unknown value: {device}')
except _pipelinewrapper.InvalidDeviceOrdinalException as e:
raise argparse.ArgumentTypeError(e)
return device
def _type_seeds(val):
try:
val = int(val)
except ValueError:
raise argparse.ArgumentTypeError('Must be an integer')
return val
def _type_gen_seeds(val):
try:
val = int(val)
except ValueError:
raise argparse.ArgumentTypeError('Must be an integer')
return dgenerate.gen_seeds(val)
def _type_animation_format(val):
val = val.lower()
if val not in _mediaoutput.supported_animation_writer_formats():
raise argparse.ArgumentTypeError(
f'Must be {_SUPPORTED_ANIMATION_OUTPUT_FORMATS_PRETTY}. Unknown value: {val}')
return val
def _type_frame_start(val):
try:
val = int(val)
except ValueError:
raise argparse.ArgumentTypeError('Must be an integer')
if val < 0:
raise argparse.ArgumentTypeError('Must be greater than or equal to 0')
return val
def _type_frame_end(val):
try:
val = int(val)
except ValueError:
raise argparse.ArgumentTypeError('Must be an integer')
if val < 0:
raise argparse.ArgumentTypeError('Must be greater than or equal to 0')
return val
def _type_image_seed_strengths(val):
try:
val = float(val)
except ValueError:
raise argparse.ArgumentTypeError('Must be a floating point number')
if val <= 0:
raise argparse.ArgumentTypeError('Must be greater than 0')
return val
def _type_upscaler_noise_levels(val):
try:
val = int(val)
except ValueError:
raise argparse.ArgumentTypeError('Must be an integer')
if val < 0:
raise argparse.ArgumentTypeError('Must be greater than or equal to 0')
return val
def _type_image_guidance_scale(val):
try:
val = float(val)
except ValueError:
raise argparse.ArgumentTypeError('Must be a floating point number')
if val < 1:
raise argparse.ArgumentTypeError('Must be greater than or equal to 1')
return val
def _type_batch_size(val):
try:
val = int(val)
except ValueError:
raise argparse.ArgumentTypeError('Must be an integer')
if val < 1:
raise argparse.ArgumentTypeError('Must be greater than or equal to 1')
return val
actions: typing.List[Action] = []
actions.append(
parser.add_argument('model_path', action='store',
help="""huggingface model repository slug, huggingface blob link to a model file,
path to folder on disk, or path to a .pt, .pth, .bin, .ckpt, or .safetensors file."""))
actions.append(
parser.add_argument('-v', '--verbose', action='store_true', default=False,
help="""Output information useful for debugging, such as pipeline
call and model load parameters."""))
actions.append(
parser.add_argument('--version', action='version', version=f"dgenerate v{dgenerate.__version__}",
help="Show dgenerate's version and exit"))
actions.append(
parser.add_argument('-pm', '--plugin-modules', action='store', default=[], nargs="+", dest='plugin_module_paths',
metavar="PATH",
help="""Specify one or more plugin module folder paths (folder containing __init__.py) or
python .py file paths to load as plugins. Plugin modules can currently only implement
image preprocessors."""))
actions.append(
parser.add_argument('-ofm', '--offline-mode', action='store_true', default=False,
help="""Whether dgenerate should try to download huggingface models that do not
exist in the disk cache, or only use what is available in the cache. Referencing
a model on huggingface that has not been cached because it was not previously
downloaded will result in a failure when using this option."""))
# This argument is handled in dgenerate.invoker.invoke_dgenerate
actions.append(
parser.add_argument('-th', '--templates-help', action='store_true', dest=None,
help="""Print a list of template variables available after a dgenerate invocation
during batch processing from STDIN. When used as a command option, their values
are not presented, just their names and types."""))
actions.append(
parser.add_argument('-mt', '--model-type', action='store', default='torch', type=_model_type,
help=f"""Use when loading different model types.
Currently supported: {_SUPPORTED_MODEL_TYPES_PRETTY}. (default: torch)"""))
actions.append(
parser.add_argument('-rev', '--revision', action='store', default="main", metavar="BRANCH",
help="""The model revision to use when loading from a huggingface repository,
(The git branch / tag, default is "main")"""))
actions.append(
parser.add_argument('-var', '--variant', action='store', default=None,
help="""If specified when loading from a huggingface repository or folder, load weights
from "variant" filename, e.g. "pytorch_model.<variant>.safetensors".
Defaults to automatic selection. This option is ignored if using flax."""))
actions.append(
parser.add_argument('-sbf', '--subfolder', action='store', default=None,
help="""Main model subfolder.
If specified when loading from a huggingface repository or folder,
load weights from the specified subfolder."""))
actions.append(
parser.add_argument('-atk', '--auth-token', action='store', default=None, metavar="TOKEN",
help="""Huggingface auth token.
Required to download restricted repositories that have access permissions
granted to your huggingface account."""))
actions.append(
parser.add_argument('-bs', '--batch-size', action='store', default=None, metavar="INTEGER", type=_type_batch_size,
help="""The number of image variations to produce per set of individual diffusion parameters
in one rendering step simultaneously on a single GPU. When using flax, batch size
is controlled by the environmental variable CUDA_VISIBLE_DEVICES which is a comma
seperated list of GPU device numbers (as listed by nvidia-smi). Usage of this
argument with --model-type flax* will cause an error, diffusion with flax will
generate an image on every GPU that is visible to CUDA and this is currently
unchangeable. When generating animations with a --batch-size greater than one,
a separate animation (with the filename suffix "animation_N") will be written to for
each image in the batch. If --batch-grid-size is specified when producing an animation
then the image grid is used for the output frames. During animation rendering each
image in the batch will still be written to the output directory along side the produced
animation as either suffixed files or image grids depending on the options you choose.
(Torch Default: 1)"""))
actions.append(
parser.add_argument('-bgs', '--batch-grid-size', action='store', default=None, metavar="SIZE", type=_type_size,
help="""Produce a single image containing a grid of images with the number of COLUMNSxROWS
given to this argument when --batch-size is greater than 1, or when using flax with multiple
GPUs visible (via the environmental variable CUDA_VISIBLE_DEVICES). If not specified with a
--batch-size greater than 1, images will be written individually with an image number suffix
(image_N) in the filename signifying which image in the batch they are."""))
actions.append(
parser.add_argument('-vae', '--vae', action='store', default=None, metavar="VAE_URI", dest='vae_uri',
help=
f"""Specify a VAE using a URI. When using torch models the URI syntax is:
"AutoEncoderClass;model=(huggingface repository slug/blob link or file/folder path)".
Examples: "AutoencoderKL;model=vae.pt", "AsymmetricAutoencoderKL;model=huggingface/vae",
"AutoencoderTiny;model=huggingface/vae".
When using a Flax model, there is currently only one available
encoder class: "FlaxAutoencoderKL;model=huggingface/vae".
The AutoencoderKL encoder class accepts huggingface repository slugs/blob links,
.pt, .pth, .bin, .ckpt, and .safetensors files. Other encoders can only accept huggingface
repository slugs/blob links, or a path to a folder on disk with the model
configuration and model file(s).
Aside from the "model" argument, there are four other optional arguments that can be specified,
these include "revision", "variant", "subfolder", "dtype".
They can be specified as so in any order, they are not positional:
"AutoencoderKL;model=huggingface/vae;revision=main;variant=fp16;subfolder=sub_folder;dtype=float16".
The "revision" argument specifies the model revision to use for the VAE when loading from
huggingface repository or blob link, (The git branch / tag, default is "main").
The "variant" argument specifies the VAE model variant, if "variant" is specified when loading
from a huggingface repository or folder, weights will be loaded from "variant" filename, e.g.
"pytorch_model.<variant>.safetensors. "variant" defaults to automatic selection and is ignored if
using flax. "variant" in the case of --vae does not default to the value of --variant to prevent
failures during common use cases.
The "subfolder" argument specifies the VAE model subfolder, if specified when loading from a
huggingface repository or folder, weights from the specified subfolder.
The "dtype" argument specifies the VAE model precision, it defaults to the value of -t/--dtype
and should be one of: {_SUPPORTED_DATA_TYPES_PRETTY}.
If you wish to load a weights file directly from disk, the simplest
way is: --vae "AutoencoderKL;my_vae.safetensors", or with a
dtype "AutoencoderKL;my_vae.safetensors;dtype=float16", all other loading arguments are unused
in this case and may produce an error message if used.
If you wish to load a specific weight file from a huggingface repository, use the blob link
loading syntax: --vae "AutoencoderKL;https://huggingface.co/UserName/repository-name/blob/main/vae_model.safetensors",
the revision argument may be used with this syntax.
"""))
actions.append(
parser.add_argument('-vt', '--vae-tiling', action='store_true', default=False,
help="""Enable VAE tiling (torch models only). Assists in the generation of
large images with lower memory overhead. The VAE will split the input tensor
into tiles to compute decoding and encoding in several steps. This is
useful for saving a large amount of memory and to allow processing larger images.
Note that if you are using --control-nets you may still run into memory
issues generating large images, or with --batch-size greater than 1."""))
actions.append(
parser.add_argument('-vs', '--vae-slicing', action='store_true', default=False,
help="""Enable VAE slicing (torch* models only). Assists in the generation
of large images with lower memory overhead. The VAE will split the input tensor
in slices to compute decoding in several steps. This is useful to save some memory,
especially when --batch-size is greater than 1. Note that if you are using --control-nets
you may still run into memory issues generating large images."""))
actions.append(
parser.add_argument('-lra', '--loras', '--lora', nargs='+', action='store', default=None, metavar="LORA_URI", dest='lora_uris',
help=
"""Specify one or more LoRA models using URIs (flax not supported). These should be a
huggingface repository slug, path to model file on disk (for example, a .pt, .pth, .bin,
.ckpt, or .safetensors file), or model folder containing model files.
huggingface blob links are not supported, see "subfolder" and "weight-name" below instead.
Optional arguments can be provided after a LoRA model specification,
these include: "scale", "revision", "subfolder", and "weight-name".
They can be specified as so in any order, they are not positional:
"huggingface/lora;scale=1.0;revision=main;subfolder=repo_subfolder;weight-name=lora.safetensors".
The "scale" argument indicates the scale factor of the LoRA.
The "revision" argument specifies the model revision to use for the VAE when loading from
huggingface repository, (The git branch / tag, default is "main").
The "subfolder" argument specifies the VAE model subfolder, if specified when loading from a
huggingface repository or folder, weights from the specified subfolder.
The "weight-name" argument indicates the name of the weights file to be loaded when
loading from a huggingface repository or folder on disk.
If you wish to load a weights file directly from disk, the simplest
way is: --lora "my_lora.safetensors", or with a scale "my_lora.safetensors;scale=1.0",
all other loading arguments are unused in this case and may produce an error message if used."""))
actions.append(
parser.add_argument('-ti', '--textual-inversions', nargs='+', action='store', default=None,
metavar="URI",
dest='textual_inversion_uris',
help=
"""Specify one or more Textual Inversion models using URIs (flax and SDXL not supported).
These should be a huggingface repository slug, path to model file on disk
(for example, a .pt, .pth, .bin, .ckpt, or .safetensors file), or model folder
containing model files.
huggingface blob links are not supported, see "subfolder" and "weight-name" below instead.
Optional arguments can be provided after the Textual Inversion model specification,
these include: "revision", "subfolder", and "weight-name".
They can be specified as so in any order, they are not positional:
"huggingface/ti_model;revision=main;subfolder=repo_subfolder;weight-name=lora.safetensors".
The "revision" argument specifies the model revision to use for the Textual Inversion model
when loading from huggingface repository, (The git branch / tag, default is "main").
The "subfolder" argument specifies the Textual Inversion model subfolder, if specified
when loading from a huggingface repository or folder, weights from the specified subfolder.
The "weight-name" argument indicates the name of the weights file to be loaded when
loading from a huggingface repository or folder on disk.
If you wish to load a weights file directly from disk, the simplest way is:
--textual-inversions "my_ti_model.safetensors", all other loading arguments
are unused in this case and may produce an error message if used."""))
actions.append(
parser.add_argument('-cn', '--control-nets', nargs='+', action='store', default=None, metavar="CONTROL_NET_URI",
dest='control_net_uris',
help=
f"""Specify one or more ControlNet models using URIs. This should be a
huggingface repository slug / blob link, path to model file on disk
(for example, a .pt, .pth, .bin, .ckpt, or .safetensors file), or model
folder containing model files. Currently all ControlNet models will
receive the same guidance image, in the future this will probably change.
Optional arguments can be provided after the ControlNet model specification, for torch
these include: "scale", "start", "end", "revision", "variant", "subfolder", and "dtype".
For flax: "scale", "revision", "subfolder", "dtype", "from_torch" (bool)
They can be specified as so in any order, they are not positional:
"huggingface/controlnet;scale=1.0;start=0.0;end=1.0;revision=main;variant=fp16;subfolder=repo_subfolder;dtype=float16".
The "scale" argument specifies the scaling factor applied to the ControlNet model,
the default value is 1.0.
The "start" (only for --model-type "torch*") argument specifies at what fraction of
the total inference steps to begin applying the ControlNet, defaults to 0.0, IE: the very beginning.
The "end" (only for --model-type "torch*") argument specifies at what fraction of
the total inference steps to stop applying the ControlNet, defaults to 1.0, IE: the very end.
The "revision" argument specifies the model revision to use for the ControlNet model
when loading from huggingface repository, (The git branch / tag, default is "main").
The "variant" (only for --model-type "torch*") argument specifies the ControlNet
model variant, if "variant" is specified when loading from a huggingface repository or folder,
weights will be loaded from "variant" filename, e.g. "pytorch_model.<variant>.safetensors. "variant"
defaults to automatic selection and is ignored if using flax. "variant" in the case of
--control-nets does not default to the value of --variant to prevent failures during common use cases.
The "subfolder" argument specifies the ControlNet model subfolder, if specified
when loading from a huggingface repository or folder, weights from the specified subfolder.
The "dtype" argument specifies the ControlNet model precision, it defaults to the value of -t/--dtype
and should be one of: {_SUPPORTED_DATA_TYPES_PRETTY}.
The "from_torch" (only for --model-type flax) this argument specifies that the ControlNet is to be
loaded and converted from a huggingface repository or file that is designed for pytorch. (Defaults to false)
If you wish to load a weights file directly from disk, the simplest way is:
--control-nets "my_controlnet.safetensors" or --control-nets "my_controlnet.safetensors;scale=1.0;dtype=float16",
all other loading arguments aside from "scale" and "dtype" are unused in this case and may produce
an error message if used ("from_torch" is available when using flax).
If you wish to load a specific weight file from a huggingface repository, use the blob link
loading syntax: --control-nets
"https://huggingface.co/UserName/repository-name/blob/main/controlnet.safetensors",
the revision argument may be used with this syntax.
"""))
_flax_scheduler_help_part = \
f' Flax schedulers: ({", ".join(e.name for e in diffusers.schedulers.FlaxKarrasDiffusionSchedulers)})' \
if _pipelinewrapper.have_jax_flax() else ''
actions.append(
parser.add_argument('-sch', '--scheduler', action='store', default=None, metavar="SCHEDULER_NAME",
help=
f'Specify a scheduler (sampler) by name. Passing "help" to this argument '
f'will print the compatible schedulers for a model without generating any images. '
f'Torch schedulers: ({", ".join(e.name for e in diffusers.schedulers.KarrasDiffusionSchedulers)}).'
+ _flax_scheduler_help_part))
actions.append(
parser.add_argument('--sdxl-refiner', action='store', default=None, metavar="MODEL_URI",
dest='sdxl_refiner_uri',
help=f"""Stable Diffusion XL (torch-sdxl) refiner model path using a URI.
This should be a huggingface repository slug / blob link, path to model file
on disk (for example, a .pt, .pth, .bin, .ckpt, or .safetensors file), or model
folder containing model files.
Optional arguments can be provided after the SDXL refiner model specification,
these include: "revision", "variant", "subfolder", and "dtype".
They can be specified as so in any order, they are not positional:
"huggingface/refiner_model_xl;revision=main;variant=fp16;subfolder=repo_subfolder;dtype=float16".
The "revision" argument specifies the model revision to use for the Textual Inversion model
when loading from huggingface repository, (The git branch / tag, default is "main").
The "variant" argument specifies the SDXL refiner model variant and defaults to the value of
--variant. When "variant" is specified when loading from a huggingface repository or folder,
weights will be loaded from "variant" filename, e.g. "pytorch_model.<variant>.safetensors.
The "subfolder" argument specifies the SDXL refiner model subfolder, if specified
when loading from a huggingface repository or folder, weights from the specified subfolder.
The "dtype" argument specifies the SDXL refiner model precision, it defaults to the value of -t/--dtype
and should be one of: {_SUPPORTED_DATA_TYPES_PRETTY}.
If you wish to load a weights file directly from disk, the simplest way is:
--sdxl-refiner "my_sdxl_refiner.safetensors" or --sdxl-refiner "my_sdxl_refiner.safetensors;dtype=float16",
all other loading arguments aside from "dtype" are unused in this case and may produce
an error message if used.
If you wish to load a specific weight file from a huggingface repository, use the blob link
loading syntax: --sdxl-refiner
"https://huggingface.co/UserName/repository-name/blob/main/refiner_model.safetensors",
the revision argument may be used with this syntax.
"""))
actions.append(
parser.add_argument('--sdxl-refiner-scheduler', action='store', default=None, metavar="SCHEDULER_NAME",
help='Specify a scheduler (sampler) by name for the SDXL refiner pass. Operates the exact'
'same way as --scheduler including the "help" option. Defaults to the value of --scheduler.'))
# SDXL Main pipeline
actions.append(
parser.add_argument('--sdxl-second-prompts', nargs='+', action='store', metavar="PROMPT",
default=None,
type=_type_prompts,
help="""One or more secondary prompts to try using SDXL's secondary text encoder.
By default the model is passed the primary prompt for this value, this option
allows you to choose a different prompt. The negative prompt component can be
specified with the same syntax as --prompts"""))
actions.append(
parser.add_argument('--sdxl-aesthetic-scores', metavar="FLOAT",
action='store', nargs='+', default=[], type=float,
help="""One or more Stable Diffusion XL (torch-sdxl) "aesthetic-score" micro-conditioning parameters.
Used to simulate an aesthetic score of the generated image by influencing the positive text
condition. Part of SDXL's micro-conditioning as explained in section 2.2 of
[https://huggingface.co/papers/2307.01952]."""))
actions.append(
parser.add_argument('--sdxl-crops-coords-top-left', metavar="COORD",
action='store', nargs='+', default=[], type=_type_image_coordinate,
help="""One or more Stable Diffusion XL (torch-sdxl) "negative-crops-coords-top-left" micro-conditioning
parameters in the format "0,0". --sdxl-crops-coords-top-left can be used to generate an image that
appears to be "cropped" from the position --sdxl-crops-coords-top-left downwards. Favorable,
well-centered images are usually achieved by setting --sdxl-crops-coords-top-left to "0,0".
Part of SDXL's micro-conditioning as explained in section 2.2 of
[https://huggingface.co/papers/2307.01952]."""))
actions.append(
parser.add_argument('--sdxl-original-size', '--sdxl-original-sizes', dest='sdxl_original_sizes', metavar="SIZE",
action='store', nargs='+', default=[], type=_type_size,
help="""One or more Stable Diffusion XL (torch-sdxl) "original-size" micro-conditioning parameters in
the format (WIDTH)x(HEIGHT). If not the same as --sdxl-target-size the image will appear to be
down or up-sampled. --sdxl-original-size defaults to --output-size or the size of any input
images if not specified. Part of SDXL\'s micro-conditioning as explained in section 2.2 of
[https://huggingface.co/papers/2307.01952]"""))
actions.append(
parser.add_argument('--sdxl-target-size', '--sdxl-target-sizes', dest='sdxl_target_sizes', metavar="SIZE",
action='store', nargs='+', default=[], type=_type_size,
help="""One or more Stable Diffusion XL (torch-sdxl) "target-size" micro-conditioning parameters in
the format (WIDTH)x(HEIGHT). For most cases, --sdxl-target-size should be set to the desired
height and width of the generated image. If not specified it will default to --output-size or
the size of any input images. Part of SDXL\'s micro-conditioning as explained in section 2.2 of
[https://huggingface.co/papers/2307.01952]"""))
actions.append(
parser.add_argument('--sdxl-negative-aesthetic-scores', metavar="FLOAT",
action='store', nargs='+', default=[], type=float,
help="""One or more Stable Diffusion XL (torch-sdxl) "negative-aesthetic-score" micro-conditioning parameters.
Part of SDXL's micro-conditioning as explained in section 2.2 of [https://huggingface.co/papers/2307.01952].
Can be used to simulate an aesthetic score of the generated image by influencing the negative text condition."""))
actions.append(
parser.add_argument('--sdxl-negative-original-sizes', metavar="SIZE",
action='store', nargs='+', default=[], type=_type_size,
help="""One or more Stable Diffusion XL (torch-sdxl) "negative-original-sizes" micro-conditioning parameters.
Negatively condition the generation process based on a specific image resolution. Part of SDXL's
micro-conditioning as explained in section 2.2 of [https://huggingface.co/papers/2307.01952].
For more information, refer to this issue thread: https://github.com/huggingface/diffusers/issues/4208"""))
actions.append(
parser.add_argument('--sdxl-negative-target-sizes', metavar="SIZE",
action='store', nargs='+', default=[], type=_type_size,
help="""One or more Stable Diffusion XL (torch-sdxl) "negative-original-sizes" micro-conditioning parameters.
To negatively condition the generation process based on a target image resolution. It should be as same
as the "--sdxl-target-size" for most cases. Part of SDXL's micro-conditioning as explained in section 2.2 of
[https://huggingface.co/papers/2307.01952]. For more information, refer to this issue thread:
https://github.com/huggingface/diffusers/issues/4208."""))
actions.append(
parser.add_argument('--sdxl-negative-crops-coords-top-left', metavar="COORD",
action='store', nargs='+', default=[], type=_type_image_coordinate,
help="""One or more Stable Diffusion XL (torch-sdxl) "negative-crops-coords-top-left" micro-conditioning
parameters in the format "0,0". Negatively condition the generation process based on a specific
crop coordinates. Part of SDXL's micro-conditioning as explained in section 2.2 of
[https://huggingface.co/papers/2307.01952]. For more information, refer
to this issue thread: https://github.com/huggingface/diffusers/issues/4208."""))
# SDXL Refiner pipeline
actions.append(
parser.add_argument('--sdxl-refiner-prompts', nargs='+', action='store',
metavar="PROMPT",
default=None,
type=_type_prompts,
help="""One or more prompts to try with the SDXL refiner model,
by default the refiner model gets the primary prompt, this argument
overrides that with a prompt of your choosing. The negative prompt
component can be specified with the same syntax as --prompts"""))
actions.append(
parser.add_argument('--sdxl-refiner-clip-skips', nargs='+', action='store', metavar="INTEGER",
default=None,
type=_type_clip_skip,
help="""One or more clip skip override values to try for the SDXL refiner,
which normally uses the clip skip value for the main model when it is
defined by --clip-skips."""))
actions.append(
parser.add_argument('--sdxl-refiner-second-prompts', nargs='+', action='store',
metavar="PROMPT",
default=None,
type=_type_prompts,
help="""One or more prompts to try with the SDXL refiner models secondary
text encoder, by default the refiner model gets the primary prompt passed
to its second text encoder, this argument overrides that with a prompt
of your choosing. The negative prompt component can be specified with the
same syntax as --prompts"""))
actions.append(
parser.add_argument('--sdxl-refiner-aesthetic-scores', metavar="FLOAT",
action='store', nargs='+', default=[], type=float,
help="See: --sdxl-aesthetic-scores, applied to SDXL refiner pass."))
actions.append(
parser.add_argument('--sdxl-refiner-crops-coords-top-left', metavar="COORD",
action='store', nargs='+', default=[], type=_type_image_coordinate,
help="See: --sdxl-crops-coords-top-left, applied to SDXL refiner pass."))
actions.append(
parser.add_argument('--sdxl-refiner-original-sizes', metavar="SIZE",
action='store', nargs='+', default=[], type=_type_size,
help="See: --sdxl-refiner-original-sizes, applied to SDXL refiner pass."))
actions.append(
parser.add_argument('--sdxl-refiner-target-sizes', metavar="SIZE",
action='store', nargs='+', default=[], type=_type_size,
help="See: --sdxl-refiner-target-sizes, applied to SDXL refiner pass."))
actions.append(
parser.add_argument('--sdxl-refiner-negative-aesthetic-scores', metavar="FLOAT",
action='store', nargs='+', default=[], type=float,
help="See: --sdxl-negative-aesthetic-scores, applied to SDXL refiner pass."))
actions.append(
parser.add_argument('--sdxl-refiner-negative-original-sizes', metavar="SIZE",
action='store', nargs='+', default=[], type=_type_size,
help="See: --sdxl-negative-original-sizes, applied to SDXL refiner pass."))
actions.append(
parser.add_argument('--sdxl-refiner-negative-target-sizes', metavar="SIZE",
action='store', nargs='+', default=[], type=_type_size,
help="See: --sdxl-negative-target-sizes, applied to SDXL refiner pass."))
actions.append(
parser.add_argument('--sdxl-refiner-negative-crops-coords-top-left', metavar="COORD",
action='store', nargs='+', default=[], type=_type_image_coordinate,
help="See: --sdxl-negative-crops-coords-top-left, applied to SDXL refiner pass."))
actions.append(
parser.add_argument('-hnf', '--sdxl-high-noise-fractions', action='store', nargs='+', default=None,
metavar="FLOAT",
type=_type_sdxl_high_noise_fractions,
help="""One or more high-noise-fraction values for Stable Diffusion XL (torch-sdxl),
this fraction of inference steps will be processed by the base model, while the rest
will be processed by the refiner model. Multiple values to this argument will result in
additional generation steps for each value. In certain situations when the mixture of denoisers
algorithm is not supported, such as when using --control-nets and inpainting with SDXL, the inverse
proportion of this value IE: (1.0 - high-noise-fraction) becomes the --image-seed-strength
input to the SDXL refiner. (default: [0.8])"""))
actions.append(
parser.add_argument('-ri', '--sdxl-refiner-inference-steps', action='store', nargs='+', default=None, metavar="INT",
type=_type_inference_steps,
help="""One or more inference steps values for the SDXL refiner when in use.
Override the number of inference steps used by the SDXL refiner,
which defaults to the value taken from --inference-steps."""))
actions.append(
parser.add_argument('-rg', '--sdxl-refiner-guidance-scales', action='store', nargs='+', default=None,
metavar="FLOAT",
type=_type_guidance_scale,
help="""One or more guidance scale values for the SDXL refiner when in use.
Override the guidance scale value used by the SDXL refiner,
which defaults to the value taken from --guidance-scales."""))
actions.append(
parser.add_argument('-rgr', '--sdxl-refiner-guidance-rescales', action='store', nargs='+', default=None,
metavar="FLOAT",
type=_type_guidance_scale,
help="""One or more guidance rescale values for the SDXL refiner when in use.
Override the guidance rescale value used by the SDXL refiner,
which defaults to the value taken from --guidance-rescales."""))
actions.append(
parser.add_argument('-sc', '--safety-checker', action='store_true', default=False,
help="""Enable safety checker loading, this is off by default.
When turned on images with NSFW content detected may result in solid black output.
Some pretrained models have no safety checker model present, in that case this
option has no effect."""))
actions.append(
parser.add_argument('-d', '--device', action='store', default='cuda', type=_type_device,
help="""cuda / cpu. (default: cuda). Use: cuda:0, cuda:1, cuda:2, etc. to specify a specific GPU.
This argument is ignored when using flax, for flax use the environmental variable
CUDA_VISIBLE_DEVICES to specify which GPUs are visible to cuda, flax will use every
visible GPU."""))
actions.append(
parser.add_argument('-t', '--dtype', action='store', default='auto', type=_type_dtype,
help=f'Model precision: {_SUPPORTED_DATA_TYPES_PRETTY}. (default: auto)'))
actions.append(
parser.add_argument('-s', '--output-size', action='store', default=None, type=_type_output_size,
metavar="SIZE",
help="""Image output size, for txt2img generation, this is the exact output size.
The dimensions specified for this value must be aligned by 8 or you will receive an error message.
If an --image-seeds URI is used its Seed, Mask, and/or Control component image sources will be
resized to this dimension with aspect ratio maintained before being used for generation by default.
Unless --no-aspect is specified, width will be fixed and a new height (aligned by 8) will be calculated
for the input images. In most cases resizing the image inputs will result in an image output of an equal
size to the inputs, except in the case of upscalers and Deep Floyd --model-type values (torch-if*). If only
one integer value is provided, that is the value for both dimensions. X/Y dimension values should be
separated by "x". This value defaults to 512x512 for Stable Diffusion when no --image-seeds are
specified (IE txt2img mode), 1024x1024 for Stable Diffusion XL (SDXL) model types, and 64x64 for
--model-type torch-if (Deep Floyd stage 1). Deep Floyd stage 1 images passed to superscaler models
(--model-type torch-ifs*) that are specified with the 'floyd' keyword argument in an --image-seeds
definition are never resized or processed in any way."""))
actions.append(
parser.add_argument('-na', '--no-aspect', action='store_true',
help="""This option disables aspect correct resizing of images provided to --image-seeds globally.
Seed, Mask, and Control guidance images will be resized to the closest dimension specified by --output-size
that is aligned by 8 pixels with no consideration of the source aspect ratio. This can be
overriden at the --image-seeds level with the image seed keyword argument 'aspect=true/false'."""))
actions.append(
parser.add_argument('-o', '--output-path', action='store', default='output',
metavar="PATH",
help="""Output path for generated images and files.
This directory will be created if it does not exist. (default: ./output)"""))
actions.append(
parser.add_argument('-op', '--output-prefix', action='store', default=None, type=str, metavar="PREFIX",
help="""Name prefix for generated images and files.
This prefix will be added to the beginning of every generated file,
followed by an underscore."""))
actions.append(
parser.add_argument('-ox', '--output-overwrite', action='store_true', default=False,
help="""Enable overwrites of files in the output directory that already exists.
The default behavior is not to do this, and instead append a filename suffix:
"_duplicate_(number)" when it is detected that the generated file name already exists."""))
actions.append(
parser.add_argument('-oc', '--output-configs', action='store_true', default=False,
help="""Write a configuration text file for every output image or animation.
The text file can be used reproduce that particular output image or animation by piping
it to dgenerate STDIN, for example "dgenerate < config.txt". These files will be written
to --output-directory and are affected by --output-prefix and --output-overwrite as well.
The files will be named after their corresponding image or animation file. Configuration
files produced for animation frame images will utilize --frame-start and --frame-end to
specify the frame number."""))
actions.append(
parser.add_argument('-om', '--output-metadata', action='store_true', default=False,
help="""Write the information produced by --output-configs to the PNG metadata of each image.
Metadata will not be written to animated files (yet). The data is written to a
PNG metadata property named DgenerateConfig and can be read using ImageMagick like so:
"magick identify -format "%%[Property:DgenerateConfig] generated_file.png"."""))
actions.append(
parser.add_argument('-p', '--prompts', nargs='+', action='store', metavar="PROMPT",
default=[_prompt.Prompt()],
type=_type_prompts,
help="""One or more prompts to try, an image group is generated for each prompt,
prompt data is split by ; (semi-colon). The first value is the positive
text influence, things you want to see. The Second value is negative
influence IE. things you don't want to see.
Example: --prompts "shrek flying a tesla over detroit; clouds, rain, missiles".
(default: [(empty string)])"""))
actions.append(
parser.add_argument('-cs', '--clip-skips', nargs='+', action='store', metavar="INTEGER",
default=None,
type=_type_clip_skip,
help="""One or more clip skip values to try. Clip skip is the number of layers to be skipped from CLIP
while computing the prompt embeddings, it must be a value greater than or equal to zero. A value of 1 means
that the output of the pre-final layer will be used for computing the prompt embeddings. This is only
supported for --model-type values "torch" and "torch-sdxl", including with --control-nets."""))
seed_options = parser.add_mutually_exclusive_group()
actions.append(
seed_options.add_argument('-se', '--seeds', nargs='+', action='store', metavar="SEED", type=_type_seeds,
help="""One or more seeds to try, define fixed seeds to achieve deterministic output.
This argument may not be used when --gse/--gen-seeds is used.
(default: [randint(0, 99999999999999)])"""))
actions.append(
parser.add_argument('-sei', '--seeds-to-images', action='store_true',
help="""When this option is enabled, each provided --seeds value or value generated by --gen-seeds
is used for the corresponding image input given by --image-seeds. If the amount of --seeds given
is not identical to that of the amount of --image-seeds given, the seed is determined as:
seed = seeds[image_seed_index %% len(seeds)], IE: it wraps around."""))
actions.append(
seed_options.add_argument('-gse', '--gen-seeds', action='store', type=_type_gen_seeds, metavar="COUNT",
dest='seeds',
help="""Auto generate N random seeds to try. This argument may not
be used when -se/--seeds is used."""))
actions.append(
parser.add_argument('-af', '--animation-format', action='store', default='mp4', type=_type_animation_format,
metavar="FORMAT",
help=f"""Output format when generating an animation from an input video / gif / webp etc.
Value must be one of: {_SUPPORTED_ANIMATION_OUTPUT_FORMATS_PRETTY}.
(default: mp4)"""))
actions.append(
parser.add_argument('-fs', '--frame-start', action='store', default=0, type=_type_frame_start,
metavar="FRAME_NUMBER",
help='Starting frame slice point for animated files, the specified frame will be included.'))
actions.append(
parser.add_argument('-fe', '--frame-end', action='store', default=None, type=_type_frame_end,
metavar="FRAME_NUMBER",
help='Ending frame slice point for animated files, the specified frame will be included.'))
actions.append(
parser.add_argument('-is', '--image-seeds', action='store', nargs='+', default=[], metavar="SEED",
help="""One or more image seed URIs to process, these may consist of URLs or file paths.
Videos / GIFs / WEBP files will result in frames being rendered as well as an animated
output file being generated if more than one frame is available in the input file.
Inpainting for static images can be achieved by specifying a black and white mask image in each
image seed string using a semicolon as the separating character, like so:
"my-seed-image.png;my-image-mask.png", white areas of the mask indicate where
generated content is to be placed in your seed image. Output dimensions specific to the
image seed can be specified by placing the dimension at the end of the string following a
semicolon like so: "my-seed-image.png;512x512" or "my-seed-image.png;my-image-mask.png;512x512".
When using --control-nets, a singular image specification is interpreted as the control
guidance image, and you can specify multiple control image sources by separating them with
commas in the case where multiple ControlNets are specified, IE:
(--image-seeds "control-image1.png, control-image2.png") OR
(--image-seeds "seed.png;control=control-image1.png, control-image2.png").
Using --control-nets with img2img or inpainting can be accomplished with the syntax:
"my-seed-image.png;mask=my-image-mask.png;control=my-control-image.png;resize=512x512".
The "mask" and "resize" arguments are optional when using --control-nets. Videos, GIFs,
and WEBP are also supported as inputs when using --control-nets, even for the "control"
argument. --image-seeds is capable of reading from multiple animated files at once or any
combination of animated files and images, the animated file with the least amount of frames
dictates how many frames are generated and static images are duplicated over the total amount
of frames. The keyword argument "aspect" can be used to determine resizing behavior when
the global argument --output-size or the local keyword argument "resize" is specified,
it is a boolean argument indicating whether aspect ratio of the input image should be
respected or ignored. The keyword argument "floyd" can be used to specify images from
a previous deep floyd stage when using --model-type torch-ifs*. When keyword arguments
are present, all applicable images such as "mask", "control", etc. must also be defined
with keyword arguments instead of with the short syntax.
"""))
image_seed_noise_opts = parser.add_mutually_exclusive_group()
actions.append(
parser.add_argument('-sip', '--seed-image-preprocessors', action='store', nargs='+', default=None,
metavar="PREPROCESSOR",
help="""Specify one or more image preprocessor actions to preform on the primary
image specified by --image-seeds. For example: --seed-image-preprocessors "flip" "mirror" "grayscale".
To obtain more information about what image preprocessors are available and how to use them,
see: --image-preprocessor-help.
"""))
actions.append(
parser.add_argument('-mip', '--mask-image-preprocessors', action='store', nargs='+', default=None,
metavar="PREPROCESSOR",
help="""Specify one or more image preprocessor actions to preform on the inpaint mask
image specified by --image-seeds. For example: --mask-image-preprocessors "invert".
To obtain more information about what image preprocessors are available and how to use them,
see: --image-preprocessor-help.
"""))
actions.append(
parser.add_argument('-cip', '--control-image-preprocessors', action='store', nargs='+', default=None,
metavar="PREPROCESSOR",
help="""Specify one or more image preprocessor actions to preform on the control
image specified by --image-seeds, this option is meant to be used with --control-nets.
Example: --control-image-preprocessors "canny;lower=50;upper=100". The delimiter "+" can be
used to specify a different preprocessor group for each image when using multiple control images with --control-nets.
For example if you have --image-seeds "img1.png, img2.png" or --image-seeds "...;control=img1.png, img2.png"
specified and multiple ControlNet models specified with --control-nets, you can specify preprocessors for
those control images with the syntax: (--control-image-preprocessors "processes-img1" + "processes-img2"),
this syntax also supports chaining of preprocessors, for example: (--control-image-preprocessors "first-process-img1"
"second-process-img1" + "process-img2"). The amount of specified preprocessors must not exceed the amount of
specified control images, or you will receive a syntax error message. Images which do not have a preprocessor
defined for them will not be preprocessed, and the plus character can be used to indicate an image is not to be
preprocessed and instead skipped over when that image is a leading element, for example
(--control-image-preprocessors + "process-second") would indicate that the first control guidance
image is not to be processed, only the second. To obtain more information about what image preprocessors
are available and how to use them, see: --image-preprocessor-help.
"""))
# This argument is handled in dgenerate.invoker.invoke_dgenerate
actions.append(
parser.add_argument('-iph', '--image-preprocessor-help', action='store', nargs='*', default=None,
metavar="PREPROCESSOR",
dest=None,
help="""Use this option alone (or with --plugin-modules) and no model
specification in order to list available image preprocessor module names.
Specifying one or more module names after this option will cause usage
documentation for the specified modules to be printed."""))
actions.append(
image_seed_noise_opts.add_argument('-iss', '--image-seed-strengths', action='store', nargs='+', default=None,
metavar="FLOAT",
type=_type_image_seed_strengths,
help=f"""One or more image strength values to try when using --image-seeds for
img2img or inpaint mode. Closer to 0 means high usage of the seed image (less noise convolution),
1 effectively means no usage (high noise convolution). Low values will produce something closer
or more relevant to the input image, high values will give the AI more creative freedom. (default: [0.8])"""))
actions.append(
image_seed_noise_opts.add_argument('-uns', '--upscaler-noise-levels', action='store', nargs='+', default=None,
metavar="INTEGER",
type=_type_upscaler_noise_levels,
help=f"""
One or more upscaler noise level values to try when using the super resolution upscaler
--model-type torch-upscaler-x4. Specifying this option for --model-type torch-upscaler-x2
will produce an error message. The higher this value the more noise is added
to the image before upscaling (similar to --image-seed-strength). (default: [20])"""))
actions.append(
parser.add_argument('-gs', '--guidance-scales', action='store', nargs='+',
default=[_pipelinewrapper.DEFAULT_GUIDANCE_SCALE],
metavar="FLOAT",
type=_type_guidance_scale,
help="""One or more guidance scale values to try. Guidance scale effects how much your
text prompt is considered. Low values draw more data from images unrelated
to text prompt. (default: [5])"""))
actions.append(
parser.add_argument('-igs', '--image-guidance-scales', action='store', nargs='+', default=None,
metavar="FLOAT",
type=_type_image_guidance_scale,
help="""One or more image guidance scale values to try. This can push the generated image towards the
initial image when using --model-type *-pix2pix models, it is unsupported for other model types.
Use in conjunction with --image-seeds, inpainting (masks) and --control-nets are not supported.
Image guidance scale is enabled by setting image-guidance-scale > 1. Higher image guidance scale
encourages generated images that are closely linked to the source image, usually at the expense of
lower image quality. Requires a value of at least 1. (default: [1.5])"""))
actions.append(
parser.add_argument('-gr', '--guidance-rescales', action='store', nargs='+', default=[],
metavar="FLOAT",
type=_type_guidance_scale,
help="""One or more guidance rescale factors to try. Proposed by [Common Diffusion Noise Schedules and
Sample Steps are Flawed](https://arxiv.org/pdf/2305.08891.pdf) "guidance_scale" is defined
as "φ" in equation 16. of [Common Diffusion Noise Schedules and Sample Steps are Flawed]
(https://arxiv.org/pdf/2305.08891.pdf). Guidance rescale factor should fix overexposure
when using zero terminal SNR. This is supported for basic text to image generation
when using --model-type "torch" but not inpainting, img2img, or --control-nets.
When using --model-type "torch-sdxl" it is supported for basic generation, inpainting,
and img2img, unless --control-nets is specified in which case only inpainting is supported.
It is supported for --model-type "torch-sdxl-pix2pix" but not --model-type "torch-pix2pix".
(default: [0.0])"""))
actions.append(
parser.add_argument('-ifs', '--inference-steps', action='store', nargs='+',
default=[_pipelinewrapper.DEFAULT_INFERENCE_STEPS],
type=_type_inference_steps,
metavar="INTEGER",
help="""One or more inference steps values to try. The amount of inference (de-noising) steps
effects image clarity to a degree, higher values bring the image closer to what
the AI is targeting for the content of the image. Values between 30-40
produce good results, higher values may improve image quality and or
change image content. (default: [30])"""))
def _type_expression(arg):
try:
_memory.memory_constraint_syntax_check(arg)
except _memory.MemoryConstraintSyntaxError as e:
raise argparse.ArgumentTypeError(f'Syntax error: {e}')
actions.append(
parser.add_argument('-mc', '--cache-memory-constraints', action='store', nargs='+',
default=None,
type=_type_expression,
metavar="EXPR",
help=f"""Cache constraint expressions describing when to clear all model caches
automatically (DiffusionPipeline, VAE, and ControlNet) considering current memory
usage. If any of these constraint expressions are met all models cached in memory will be cleared.
Example, and default value: {' '.join(_textprocessing.quote_spaces(_pipelinewrapper.CACHE_MEMORY_CONSTRAINTS))}"""
f' For Syntax See: [https://dgenerate.readthedocs.io/en/v{dgenerate.__version__}/'
f'dgenerate_submodules.html#dgenerate.pipelinewrapper.CACHE_MEMORY_CONSTRAINTS]'))
actions.append(
parser.add_argument('-pmc', '--pipeline-cache-memory-constraints', action='store', nargs='+',
default=None,
type=_type_expression,
metavar="EXPR",
help=f"""Cache constraint expressions describing when to automatically clear the in memory
DiffusionPipeline cache considering current memory usage, and estimated memory usage of
new models that are about to enter memory. If any of these constraint expressions are
met all DiffusionPipeline objects cached in memory will be cleared. Example, and default
value: {' '.join(_textprocessing.quote_spaces(_pipelinewrapper.PIPELINE_CACHE_MEMORY_CONSTRAINTS))}"""
f' For Syntax See: [https://dgenerate.readthedocs.io/en/v{dgenerate.__version__}/'
f'dgenerate_submodules.html#dgenerate.pipelinewrapper.PIPELINE_CACHE_MEMORY_CONSTRAINTS]'))
actions.append(
parser.add_argument('-vmc', '--vae-cache-memory-constraints', action='store', nargs='+',
default=None,
type=_type_expression,
metavar="EXPR",
help=f"""Cache constraint expressions describing when to automatically clear the in memory VAE
cache considering current memory usage, and estimated memory usage of new VAE models that
are about to enter memory. If any of these constraint expressions are met all VAE
models cached in memory will be cleared. Example, and default
value: {' '.join(_textprocessing.quote_spaces(_pipelinewrapper.VAE_CACHE_MEMORY_CONSTRAINTS))}"""
f' For Syntax See: [https://dgenerate.readthedocs.io/en/v{dgenerate.__version__}/'
f'dgenerate_submodules.html#dgenerate.pipelinewrapper.VAE_CACHE_MEMORY_CONSTRAINTS]'))
actions.append(
parser.add_argument('-cmc', '--control-net-cache-memory-constraints', action='store', nargs='+',
default=None,
type=_type_expression,
metavar="EXPR",
help=f"""Cache constraint expressions describing when to automatically clear the in memory ControlNet
cache considering current memory usage, and estimated memory usage of new ControlNet models that
are about to enter memory. If any of these constraint expressions are met all ControlNet
models cached in memory will be cleared. Example, and default
value: {' '.join(_textprocessing.quote_spaces(_pipelinewrapper.CONTROL_NET_CACHE_MEMORY_CONSTRAINTS))}"""
f' For Syntax See: [https://dgenerate.readthedocs.io/en/v{dgenerate.__version__}/'
f'dgenerate_submodules.html#dgenerate.pipelinewrapper.CONTROL_NET_CACHE_MEMORY_CONSTRAINTS]'))
[docs]
class DgenerateUsageError(Exception):
pass
[docs]
class DgenerateArguments(dgenerate.RenderLoopConfig):
"""
Represents dgenerates parsed command line arguments, can be used
as a configuration object for :py:class:`dgenerate.renderloop.RenderLoop`.
"""
plugin_module_paths: _types.Paths
verbose: bool = False
cache_memory_constraints: typing.Optional[typing.List[str]] = None
pipeline_cache_memory_constraints: typing.Optional[typing.List[str]] = None
vae_cache_memory_constraints: typing.Optional[typing.List[str]] = None
control_net_cache_memory_constraints: typing.Optional[typing.List[str]] = None
[docs]
def __init__(self):
super().__init__()
self.plugin_module_paths = []
_attr_name_to_option = {a.dest: a.option_strings[-1] if a.option_strings else a.dest for a in actions}
[docs]
def config_attribute_name_to_option(name):
"""
Convert an attribute name of :py:class:`.DgenerateArguments` into its command line option name.
:param name: the attribute name
:return: the command line argument name as a string
"""
return _attr_name_to_option[name]
def _parse_args(args=None) -> DgenerateArguments:
args = parser.parse_args(args, namespace=DgenerateArguments())
args.check(config_attribute_name_to_option)
return args
[docs]
def parse_args(args: typing.Sequence[str],
throw: bool = True) -> typing.Union[DgenerateArguments, None]:
"""
Parse dgenerates command line arguments and return a configuration object.
:param args: arguments list, as in args taken from sys.argv, or in that format
:param throw: throw :py:exc:`.DgenerateUsageError` on error? defaults to True
:raise DgenerateUsageError:
:return: :py:class:`.DgenerateArguments`. If ``throw=False`` then
``None`` will be returned on errors.
"""
try:
return _parse_args(args)
except (dgenerate.RenderLoopConfigError, argparse.ArgumentTypeError, argparse.ArgumentError) as e:
_messages.log(f'dgenerate: error: {e}', level=_messages.ERROR)
if throw:
raise DgenerateUsageError(e)
return None