"git@developer.sourcefind.cn:OpenDAS/megatron-lm.git" did not exist on "160bf237afefbd4a123bccee1ff7a5e98d04974f"
Unverified Commit bc287324 authored by Nicolas Patry's avatar Nicolas Patry Committed by GitHub
Browse files

fix(launcher): Rename `b-float16` to `bfloat16` in the launcher arg (#621)

parent a2cf1bdb
...@@ -41,6 +41,7 @@ impl std::fmt::Display for Quantization { ...@@ -41,6 +41,7 @@ impl std::fmt::Display for Quantization {
#[derive(Clone, Copy, Debug, ValueEnum)] #[derive(Clone, Copy, Debug, ValueEnum)]
enum Dtype { enum Dtype {
Float16, Float16,
#[clap(name = "bfloat16")]
BFloat16, BFloat16,
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment