We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 6c0fa6b commit 7bb5ca6Copy full SHA for 7bb5ca6
1 file changed
examples/diffusers/quantization/quantize.py
@@ -340,7 +340,7 @@ def create_argument_parser() -> argparse.ArgumentParser:
340
model_group.add_argument(
341
"--model-dtype",
342
type=str,
343
- default="BFloat16",
+ default="Half",
344
choices=[d.value for d in DataType],
345
help="Precision for loading the pipeline. If you want different dtypes for separate components, "
346
"please specify using --component-dtype",
0 commit comments