Skip to content

Commit 7bb5ca6

Browse files
committed
Update the default model type
Signed-off-by: Jingyu Xin <jingyux@nvidia.com>
1 parent 6c0fa6b commit 7bb5ca6

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

examples/diffusers/quantization/quantize.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -340,7 +340,7 @@ def create_argument_parser() -> argparse.ArgumentParser:
340340
model_group.add_argument(
341341
"--model-dtype",
342342
type=str,
343-
default="BFloat16",
343+
default="Half",
344344
choices=[d.value for d in DataType],
345345
help="Precision for loading the pipeline. If you want different dtypes for separate components, "
346346
"please specify using --component-dtype",

0 commit comments

Comments
 (0)