override dtype with quant (#1062)

This commit is contained in:
Awni Hannun
2024-10-22 09:56:45 -07:00
committed by GitHub
parent 743763bc2e
commit 66e7bcb886
3 changed files with 3 additions and 3 deletions

View File

@@ -31,7 +31,7 @@ def configure_parser() -> argparse.ArgumentParser:
)
parser.add_argument(
"--dtype",
help="Type to save the parameters, ignored if -q is given.",
help="Type to save the non-quantized parameters.",
type=str,
choices=["float16", "bfloat16", "float32"],
default="float16",