Skip to content

Commit a12a157

Browse files
committed
update --quantizer and --quantizer-map argument help
1 parent 78c561e commit a12a157

File tree

3 files changed

+19
-15
lines changed

3 files changed

+19
-15
lines changed

dgenerate/arguments.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1930,11 +1930,13 @@ class names that indicates what YOLO detection classes to keep. This filter is a
19301930
This URI specifies the quantization backend and its configuration.
19311931
19321932
Quantization will be applied to all text encoders, and unet / transformer
1933-
models with the provided settings when using this argument.
1933+
models with the provided settings when using this argument. ControlNet models
1934+
are NOT quantized by default and must be explicitly included via --quantizer-map.
19341935
1935-
If you wish to specify different quantization types per encoder or unet / transformer,
1936-
you should use the "quantizer" URI argument of --text-encoders and or --unet / --transformer
1937-
to specify the quantization settings on a per model basis.
1936+
If you wish to specify different quantization types per encoder, unet / transformer,
1937+
or controlnet, you should use the "quantizer" URI argument of --text-encoders,
1938+
--unet / --transformer, or --control-nets to specify the quantization settings
1939+
on a per model basis.
19381940
19391941
Available backends are: (bnb / bitsandbytes, sdnq)
19401942
@@ -2008,12 +2010,12 @@ class names that indicates what YOLO detection classes to keep. This filter is a
20082010
performed on them.
20092011
20102012
By default when a --quantizer URI is specified, the UNet / Transformer, and all Text Encoders
2011-
are processed.
2013+
are processed. ControlNet models are NOT processed by default.
20122014
20132015
When using --quantizer, you can use this argument to specify exactly which sub-modules undergo
20142016
quantization.
20152017
2016-
Accepted values are: "unet", "transformer", "text_encoder", "text_encoder_2", "text_encoder_3"
2018+
Accepted values are: "unet", "transformer", "text_encoder", "text_encoder_2", "text_encoder_3", "controlnet"
20172019
"""
20182020
)
20192021
)

dgenerate/console/schemas/arguments.json

Lines changed: 1 addition & 1 deletion
Large diffs are not rendered by default.

docs/manual.rst

Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -930,11 +930,12 @@ Help Output
930930
This URI specifies the quantization backend and its configuration.
931931

932932
Quantization will be applied to all text encoders, and unet / transformer models with the provided
933-
settings when using this argument.
933+
settings when using this argument. ControlNet models are NOT quantized by default and must be
934+
explicitly included via --quantizer-map.
934935

935-
If you wish to specify different quantization types per encoder or unet / transformer, you should
936-
use the "quantizer" URI argument of --text-encoders and or --unet / --transformer to specify the
937-
quantization settings on a per model basis.
936+
If you wish to specify different quantization types per encoder, unet / transformer, or controlnet,
937+
you should use the "quantizer" URI argument of --text-encoders, --unet / --transformer, or
938+
--control-nets to specify the quantization settings on a per model basis.
938939

939940
Available backends are: (bnb / bitsandbytes, sdnq)
940941

@@ -989,13 +990,14 @@ Help Output
989990
on them.
990991

991992
By default when a --quantizer URI is specified, the UNet / Transformer, and all Text Encoders are
992-
processed.
993+
processed. ControlNet models are NOT processed by default.
993994

994995
When using --quantizer, you can use this argument to specify exactly which sub-modules undergo
995996
quantization.
996997

997-
Accepted values are: "unet", "transformer", "text_encoder", "text_encoder_2", "text_encoder_3"
998-
----------------------------------------------------------------------------------------------
998+
Accepted values are: "unet", "transformer", "text_encoder", "text_encoder_2", "text_encoder_3",
999+
"controlnet"
1000+
------------
9991001
-q2, --second-model-quantizer QUANTIZER_URI
10001002
Global quantization configuration via URI for the secondary model, such as the SDXL Refiner or
10011003
Stable Cascade decoder. See: --quantizer for syntax examples.
@@ -10612,7 +10614,7 @@ The ``\templates_help`` output from the above example is:
1061210614
Value: []
1061310615
Name: "last_seeds"
1061410616
Type: collections.abc.Sequence[int]
10615-
Value: [59460199418710]
10617+
Value: [46306913169477]
1061610618
Name: "last_seeds_to_images"
1061710619
Type: <class 'bool'>
1061810620
Value: False

0 commit comments

Comments
 (0)